Views
No views yet
No system prompt needed. VYOM identity is fine-tuned in.
1from transformers import AutoTokenizer, AutoModelForCausalLM
2import torch
3
4tokenizer = AutoTokenizer.from_pretrained("anilsuthar2004/VYOM_4B")
5model = AutoModelForCausalLM.from_pretrained(
6 "anilsuthar2004/VYOM_4B",
7 torch_dtype=torch.float16,
8 device_map="auto"
9)
10
11# No system prompt needed — persona is in the weights!
12messages = [{"role": "user", "content": "Tu kaun hai?"}]
13text = tokenizer.apply_chat_template(
14 messages, tokenize=False,
15 add_generation_prompt=True,
16 enable_thinking=False
17)
18inputs = tokenizer(text, return_tensors="pt").to(model.device)
19out = model.generate(
20 **inputs,
21 max_new_tokens=200,
22 temperature=0.7,
23 top_p=0.8,
24 top_k=20,
25 repetition_penalty=1.0
26)
27print(tokenizer.decode(
28 out[0][inputs["input_ids"].shape[1]:],
29 skip_special_tokens=True
30))| Field | Value |
|---|---|
| Method | LoRA (16-bit, not QLoRA) |
| LoRA rank | 16 |
| LoRA alpha | 32 |
| Epochs | 5 |
| Batch size | 4 (2 per GPU × 2 T4s) |
| Learning rate | 2e-4 |
| Scheduler | cosine |
| Optimizer | adamw_8bit |
| Platform | Kaggle 2×T4 |
| Seq length | 2048 |
| Training data | 60 VYOM Q&A pairs (Hinglish) |
| Parameter | Value |
|---|---|
| temperature | 0.7 |
| top_p | 0.8 |
| top_k | 20 |
| repetition_penalty | 1.0 |