keivalya/MedQuad-MedicalQnADataset
Viewer • Updated • 16.4k • 4.95k • 132
How to use Arushp1/llama3-medquad-qlora with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("question-answering", model="Arushp1/llama3-medquad-qlora") # Load model directly
from transformers import AutoModel
model = AutoModel.from_pretrained("Arushp1/llama3-medquad-qlora", device_map="auto")This model is a fine-tuned version of LLaMA-3 8B Instruct using QLoRA (4-bit quantization + LoRA adapters) on the MedQuad Medical QnA Dataset.
It is designed to answer medical domain questions across various categories like treatment, symptoms, causes, prevention, inheritance, etc.
⚠️ This model is not a substitute for professional medical advice. It should not be used for clinical decision-making or diagnosis.
'''https://huggingface.co/Arushp1/llama3-medquad-qlora'''
from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
model = AutoModelForCausalLM.from_pretrained("Arushp1/llama3-medquad-qlora")
tokenizer = AutoTokenizer.from_pretrained("Arushp1/llama3-medquad-qlora")
pipe = pipeline("text-generation", model=model, tokenizer=tokenizer)
query = "What are the symptoms of asthma?"
print(pipe(query, max_new_tokens=100))
Base model
meta-llama/Llama-3.1-8B