Views
No views yet
ChatDoctor-HealthCareMagic-100k dataset to provide empathetic, doctor-style responses to medical queries.unsloth/llama-3-8b-instruct-bnb-4bit1from peft import PeftModel, PeftConfig
2from transformers import AutoModelForCausalLM, AutoTokenizer
3
4# 1. Load Base Model
5base_model_name = "unsloth/llama-3-8b-instruct-bnb-4bit"
6base_model = AutoModelForCausalLM.from_pretrained(
7 base_model_name,
8 load_in_4bit=True,
9 device_map="auto"
10)
11tokenizer = AutoTokenizer.from_pretrained(base_model_name)
12
13# 2. Load HACK_DOC Adapter
14model = PeftModel.from_pretrained(base_model, "shri171981/genai_hack_doc")
15
16# 3. Run Inference
17inputs = tokenizer("I have a severe headache.", return_tensors="pt").to("cuda")
18outputs = model.generate(**inputs, max_new_tokens=128)
19print(tokenizer.decode(outputs[0], skip_special_tokens=True))