Update README.md

polm-stability · Mar 30, 2023 · 0f84a60 · 0f84a60
1 parent 56ae305
commit 0f84a60
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/README.md b/README.md
@@ -59,6 +59,7 @@ target_embeddings, info = wechsel.apply(
 )
 
 model.get_input_embeddings().weight.data = torch.from_numpy(target_embeddings)
+model.config.vocab_size = len(target_embeddings)
 
 # use `model` and `target_tokenizer` to continue training in Swahili!
 ```