Views
No views yet
1from transformers import AutoModelForCausalLM, AutoTokenizer
2import torch
3
4model = AutoModelForCausalLM.from_pretrained("MURUGAYESU/tamil-govt-scheme-llama3-8b", torch_dtype=torch.float16, device_map='auto')
5tokenizer = AutoTokenizer.from_pretrained("MURUGAYESU/tamil-govt-scheme-llama3-8b")
6
7prompt = '''### Instruction:
8கீழே கொடுக்கப்பட்ட அரசு திட்டம் பற்றிய கேள்விக்கு தமிழில் விரிவான பதிலளிக்கவும்.
9
10### Input:
11PM-KISAN திட்டத்திற்கு யார் தகுதியானவர்கள்?
12
13### Response:
14'''
15inputs = tokenizer(prompt, return_tensors='pt').to('cuda')
16output = model.generate(**inputs, max_new_tokens=150, temperature=0.1, do_sample=True)
17print(tokenizer.decode(output[0][inputs['input_ids'].shape[1]:], skip_special_tokens=True))