Meditron 7b Chat

🚀 meditron-7b-chat

meditron-7b-chat is a fine - tuned model based on epfl - llm/meditron - 7b, which can answer various medical questions.

🚀 Quick Start

Basic Usage

from transformers import AutoTokenizer,AutoModelForCausalLM
 
model_id = "malhajar/meditron-7b-chat"
model = AutoModelForCausalLM.from_pretrained(model_name_or_path,
                                             device_map="auto",
                                             torch_dtype=torch.float16,
                                             revision="main")

tokenizer = AutoTokenizer.from_pretrained(model_id)

question: "what is tract infection?"
# For generating a response
prompt = '''
### Instruction:
{question} 

### Response:'''
input_ids = tokenizer(prompt, return_tensors="pt").input_ids
output = model.generate(inputs=input_ids,max_new_tokens=512,pad_token_id=tokenizer.eos_token_id,top_k=50, do_sample=True,
        top_p=0.95)
response = tokenizer.decode(output[0])

print(response)

📚 Documentation

Model Description

• Finetuned by: Mohamad Alhajar
• Language(s) (NLP): English
• Finetuned from model: epfl-llm/meditron-7b

Prompt Template

### Instruction:

<prompt> (without the <>)

### Response:

📊 Evaluation Results

Open LLM Leaderboard Evaluation Results

Detailed results can be found here

📄 License

The model uses the Llama2 license.