Views
No views yet
1from unsloth import FastLanguageModel
2import torch
3
4model, tokenizer = FastLanguageModel.from_pretrained(
5 model_name="ThakrePranjal/pharma-tinyllama-unsloth-stage2-merged",
6 max_seq_length=512,
7 load_in_4bit=True,
8)
9FastLanguageModel.for_inference(model)
10
11prompt = "### Instruction:\nExplain the mechanism of metformin.\n\n### Response:\n"
12inputs = tokenizer(prompt, return_tensors="pt").to("cuda")
13with torch.inference_mode():
14 out = model.generate(**inputs, max_new_tokens=150, temperature=0.7,
15 top_p=0.9, do_sample=True,
16 pad_token_id=tokenizer.eos_token_id)
17print(tokenizer.decode(out[0], skip_special_tokens=True))unsloth/tinyllama-bnb-4bit
└── Stage 1 → [ThakrePranjal/pharma-tinyllama-unsloth-stage1-merged]
└── Stage 2 SFT LoRA → Merge → THIS MODEL
└── Stage 3 DPO → [ThakrePranjal/pharma-tinyllama-unsloth-final]