Views
No views yet
⚠️ DEPRECATED VERSIONPlease use V8.5.3 instead.This model has known issues and is kept for research purposes only. V8.5.3 fixes all bugs and is the current production model.
facebook/nllb-200-distilled-600M for English → Bishnupriya Manipuri.1from peft import PeftModel
2from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
3import torch
4
5base = AutoModelForSeq2SeqLM.from_pretrained("facebook/nllb-200-distilled-600M")
6model = PeftModel.from_pretrained(base, "Emarthar/nllb-bpy-beng-v8_4")
7tokenizer = AutoTokenizer.from_pretrained("Emarthar/nllb-bpy-beng-v8_4")
8model.eval()
9
10def translate(text):
11 tokenizer.src_lang = "eng_Latn"
12 inputs = tokenizer(text, return_tensors="pt")
13 with torch.no_grad():
14 out = model.generate(
15 **inputs,
16 forced_bos_token_id=tokenizer.convert_tokens_to_ids("asm_Beng"),
17 max_new_tokens=64,
18 num_beams=5
19 )
20 return tokenizer.batch_decode(out, skip_special_tokens=True)[0]
21
22print(translate("Water is important")) # পানীহান দরকারি
23print(translate("The sky is blue")) # হাগহান নীলুৱাহান
24print(translate("My name is Arunita")) # মর নাংহান অরুনিতা