Views
No views yet
facebook/nllb-200-distilled-600M1from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
2from peft import PeftModel
3
4base_model_name = "facebook/nllb-200-distilled-600M"
5adapter_path = "TajikNLPWorld/nllb-600m-tajik-persian-lora"
6
7tokenizer = AutoTokenizer.from_pretrained(base_model_name)
8model = AutoModelForSeq2SeqLM.from_pretrained(base_model_name, device_map="auto")
9model = PeftModel.from_pretrained(model, adapter_path)
10model = model.merge_and_unload() # optional
11
12# Translate
13tokenizer.src_lang = "tg_Cyrl"
14text = "ришк"
15inputs = tokenizer(text, return_tensors="pt")
16outputs = model.generate(**inputs, max_length=64)
17print(tokenizer.decode(outputs[0], skip_special_tokens=True))adapter_config.json – LoRA configuration.adapter_model.bin – LoRA weights.results/ – Folder containing evaluation metrics, plots, and predictions.