ai-forever
/

RuM2M100-1.2B

Text2Text Generation

natural language generation

Inference Endpoints

Model card Files Files and versions Community

Tokenizer loading fix

#1

by Den4ikAI - opened Oct 22, 2023

base: refs/heads/main

←

from: refs/pr/1

Discussion Files changed

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -91,7 +91,7 @@ from transformers import M2M100ForConditionalGeneration, M2M100Tokenizer
 path_to_model = "ai-forever/RuM2M100-1.2B"
 model = M2M100ForConditionalGeneration.from_pretrained(path_to_model)
-tokenizer = M2M100Tokenizer.from_pretrained(path_to_model)
 sentence = "прийдя в МГТУ я был удивлен никого необноружив там…"

 path_to_model = "ai-forever/RuM2M100-1.2B"
 model = M2M100ForConditionalGeneration.from_pretrained(path_to_model)
+tokenizer = M2M100Tokenizer.from_pretrained(path_to_model, src_lang="ru", tgt_lang="ru")
 sentence = "прийдя в МГТУ я был удивлен никого необноружив там…"