Meditron 7b Chat

🚀 meditron-7b-chat模型

meditron-7b-chat是基於epfl-llm/meditron-7b模型，在Alpaca數據集上進行SFT訓練得到的微調版本。該模型能夠回答醫學領域不同明確概念的相關信息（更多信息請參考epfl-llm/meditron-7b）。

🚀 快速開始

使用原文章中提供的代碼示例與模型進行交互。

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)

✨ 主要特性

模型描述

• 微調者：Mohamad Alhajar
• 語言（NLP）：英語
• 微調基礎模型：epfl-llm/meditron-7b

提示模板

### Instruction:

<prompt> (without the <>)

### Response:

📚 詳細文檔

Open LLM Leaderboard評估結果

詳細結果可查看此處

📄 許可證

該模型使用Llama2許可證。

📦 模型信息