meditron-7b-chat is a finetuned version of 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)
### Instruction:
<prompt> (without the <>)
### Response:
How to Get Started with the Model
Use the code sample provided in the original post to interact with the model.
fromtransformersimportAutoTokenizer,AutoModelForCausalLMmodel_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 responseprompt='''
### Instruction:
{question}### Response:'''input_ids=tokenizer(prompt,return_tensors="pt").input_idsoutput=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)