Views
No views yet
epfl-llm/meditron-7b using SFT Training on the Alpaca Dataset.
This model can answer information about different excplicit ideas in medicine (see epfl-llm/meditron-7b for more info)Mohamad Alhajarepfl-llm/meditron-7b### Instruction:
<prompt> (without the <>)
### Response:1from transformers import AutoTokenizer,AutoModelForCausalLM
2
3model_id = "malhajar/meditron-7b-chat"
4model = AutoModelForCausalLM.from_pretrained(model_name_or_path,
5 device_map="auto",
6 torch_dtype=torch.float16,
7 revision="main")
8
9tokenizer = AutoTokenizer.from_pretrained(model_id)
10
11question: "what is tract infection?"
12# For generating a response
13prompt = '''
14### Instruction:
15{question}
16
17### Response:'''
18input_ids = tokenizer(prompt, return_tensors="pt").input_ids
19output = model.generate(inputs=input_ids,max_new_tokens=512,pad_token_id=tokenizer.eos_token_id,top_k=50, do_sample=True,
20 top_p=0.95)
21response = tokenizer.decode(output[0])
22
23print(response)| Metric | Value |
|---|---|
| Avg. | 49.59 |
| AI2 Reasoning Challenge (25-Shot) | 50.77 |
| HellaSwag (10-Shot) | 75.37 |
| MMLU (5-Shot) | 40.49 |
| TruthfulQA (0-shot) | 48.56 |
| Winogrande (5-shot) | 73.16 |
| GSM8k (5-shot) | 9.17 |