Views
No views yet
meta-llama/Llama-3.2-3B-Instruct or microsoft/Phi-3.5-mini-instruct) trained on Spanish Patient Leaflets (Prospectos) and Summaries of Product Characteristics (Fichas Técnicas) from the AEMPS CIMA Research Dataset (hsilvosa/aemps-cima) and (hsilvosa/openplacsp).1from transformers import AutoTokenizer, AutoModelForCausalLM
2from peft import PeftModel
3
4base_model = "meta-llama/Llama-3.2-3B-Instruct"
5tokenizer = AutoTokenizer.from_pretrained(base_model)
6model = AutoModelForCausalLM.from_pretrained(base_model, torch_dtype="auto", device_map="auto")
7model = PeftModel.from_pretrained(model, "your-hf-username/CIMA-Spanish-Medical-Llama-LoRA")
8
9prompt = "<|system|>
10Eres un asistente médico farmacéutico...
11<|user|>
12¿Cuál es la vía de administración y composición de Omeprazol?
13<|assistant|>
14"
15inputs = tokenizer(prompt, return_tensors="pt").to("cuda")
16outputs = model.generate(**inputs, max_new_tokens=150)
17print(tokenizer.decode(outputs[0], skip_special_tokens=True))