Views
No views yet
1import torch
2from transformers import AutoModelForCausalLM, AutoTokenizer
3from peft import PeftModel
4
5base_id = "deepseek-ai/DeepSeek-R1-Distill-Qwen-1.5B"
6adapter_id = "prithvi1029/deepseek-medquad-lora"
7
8tok = AutoTokenizer.from_pretrained(base_id, trust_remote_code=True)
9model = AutoModelForCausalLM.from_pretrained(
10 base_id, device_map='auto', torch_dtype=torch.float16, trust_remote_code=True
11)
12model = PeftModel.from_pretrained(model, adapter_id)
13
14prompt = "Question: What is diabetic retinopathy?\nAnswer:"
15inputs = tok(prompt, return_tensors='pt').to(model.device)
16out = model.generate(**inputs, max_new_tokens=128)
17print(tok.decode(out[0], skip_special_tokens=True))