Views
No views yet
| Metric | Score |
|---|---|
| Train loss | 0.3375 |
| Eval loss | 0.2073 |
| Perplexity | 1.15 |
| ROUGE-1 | 0.1762 |
| ROUGE-2 | 0.1501 |
| ROUGE-L | 0.1754 |
| Train samples | 10,000 |
1from peft import PeftModel
2from transformers import AutoModelForCausalLM, AutoTokenizer
3
4base = AutoModelForCausalLM.from_pretrained("Qwen/Qwen2.5-1.5B-Instruct")
5model = PeftModel.from_pretrained(base, "Subh24ai/yojana-sahayak-qwen2.5-1.5b-qlora")
6tokenizer = AutoTokenizer.from_pretrained("Subh24ai/yojana-sahayak-qwen2.5-1.5b-qlora")
7
8messages = [
9 {"role": "system", "content": "You are Yojana Sahayak..."},
10 {"role": "user", "content": "PM Kisan ke liye kaun eligible hai?"}
11]
12prompt = tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
13inputs = tokenizer(prompt, return_tensors="pt")
14output = model.generate(**inputs, max_new_tokens=200)
15print(tokenizer.decode(output[0], skip_special_tokens=True))