Views
No views yet
1from transformers import AutoModelForCausalLM, AutoTokenizer
2tok = AutoTokenizer.from_pretrained("skyblanket/tamil-800m-instruct")
3model = AutoModelForCausalLM.from_pretrained("skyblanket/tamil-800m-instruct", dtype="bfloat16", device_map="auto")
4ids = tok("இந்தியாவின் தலைநகரம் என்ன?", return_tensors="pt").to(model.device)
5print(tok.decode(model.generate(**ids, max_new_tokens=100, do_sample=True, temperature=0.7, top_p=0.9, repetition_penalty=1.2)[0]))