Views
No views yet
facebook/nllb-200-distilled-600M:
...facebook/nllb-200-distilled-600M:1from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
2from peft import PeftModel
3import torch
4
5BASE_MODEL = "facebook/nllb-200-distilled-600M"
6LORA_REPO = "frankiethiak/nllb-mizo-bible-lora" # update if different
7
8tokenizer = AutoTokenizer.from_pretrained(
9 BASE_MODEL,
10 src_lang="eng_Latn",
11 tgt_lang="lus_Latn",
12)
13
14base_model = AutoModelForSeq2SeqLM.from_pretrained(BASE_MODEL)
15model = PeftModel.from_pretrained(base_model, LORA_REPO)
16model.eval()
17
18text = "In the beginning God created the heaven and the earth."
19inputs = tokenizer(text, return_tensors="pt")
20with torch.no_grad():
21 gen = model.generate(**inputs, max_new_tokens=80, num_beams=4)
22
23print(tokenizer.decode(gen[0], skip_special_tokens=True))