Medichat V2 Llama3 8B

🚀 Medichat-V2-Llama3-8B

Medichat-V2-Llama3-8B 是一个基于预训练语言模型融合而成的模型，借助 mergekit 工具创建。该模型在处理非结构化临床文本的结构化任务上表现出色，能有效提升医疗文本处理的效率和准确性。

✨ 主要特性

模型构成与特点

1. 基础模型：此模型以 "refuelai/Llama-3-Refueled" 为基础，它是 Llama3 - 8B 模型的优化版本，具备出色的指令遵循能力，能灵活适应不同领域的任务。
2. 融合模型：
   • ruslanmv/ai - medical - model - 32bit：专门针对医学技术问题进行微调训练，为模型提供了坚实的医学知识基础。
   • Locutusque/Llama - 3 - Hercules - 5.0 - 8B：擅长遵循复杂指令，在科学和技术对话场景中表现优异，能有效处理各类对话交互。

融合方法

本模型采用 DARE TIES 融合方法，以 [refuelai/Llama - 3 - Refueled](https://huggingface.co/refuelai/Llama - 3 - Refueled) 为基础进行融合。

融合模型列表

以下模型参与了融合：

• [ruslanmv/ai - medical - model - 32bit](https://huggingface.co/ruslanmv/ai - medical - model - 32bit)
• [Locutusque/Llama - 3 - Hercules - 5.0 - 8B](https://huggingface.co/Locutusque/Llama - 3 - Hercules - 5.0 - 8B)

配置信息

以下是用于生成此模型的 YAML 配置：

models:
  - model: Locutusque/Llama-3-Hercules-5.0-8B
    parameters:
      weight: [0.25, 0.35, 0.45, 0.35, 0.25]
      density: [0.1, 0.25, 0.5, 0.25, 0.1]
  - model: refuelai/Llama-3-Refueled
  - model: ruslanmv/ai-medical-model-32bit
    parameters:
      weight: [0.55, 0.45, 0.35, 0.45, 0.55]
      density: [0.1, 0.25, 0.5, 0.25, 0.1]
merge_method: dare_ties
base_model: refuelai/Llama-3-Refueled
parameters:
  int8_mask: true
dtype: bfloat16

📦 安装指南

文档未提供具体安装步骤，可参考相关依赖库（如 transformers）的官方文档进行安装。

💻 使用示例

基础用法

import torch
from transformers import AutoTokenizer, AutoModelForCausalLM

class MedicalAssistant:
    def __init__(self, model_name="sethuiyer/Medichat-V2-Llama3-8B", device="cuda"):
        self.device = device
        self.tokenizer = AutoTokenizer.from_pretrained(model_name)
        self.model = AutoModelForCausalLM.from_pretrained(model_name).to(self.device)
        self.sys_message = ''' 
        You are an AI Medical Assistant trained on a vast dataset of health information. Please be thorough and
        provide an informative answer. If you don't know the answer to a specific medical inquiry, advise seeking professional help.
        '''

    def format_prompt(self, question):
        messages = [
            {"role": "system", "content": self.sys_message},
            {"role": "user", "content": question}
        ]
        prompt = self.tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
        return prompt

    def generate_response(self, question, max_new_tokens=512):
        prompt = self.format_prompt(question)
        inputs = self.tokenizer(prompt, return_tensors="pt").to(self.device)
        with torch.no_grad():
            outputs = self.model.generate(**inputs, max_new_tokens=max_new_tokens, use_cache=True)
        answer = self.tokenizer.batch_decode(outputs, skip_special_tokens=True)[0].strip()
        return answer

assistant = MedicalAssistant()
question = '''
Symptoms:
Dizziness, headache, and nausea.

What is the differential diagnosis?
'''
response = assistant.generate_response(question)
print(response)

Based on the symptoms of dizziness, headache, and nausea, the differential diagnosis could include several conditions. Here are some possible causes:

1. Vestibular disorders: These can cause dizziness, vertigo, and nausea. Examples include benign paroxysmal positional vertigo (BPPV), labyrinthitis, and vestibular neuritis.

2. Migraine: Migraines can cause severe headaches, nausea, and dizziness. They can be triggered by various factors, such as stress, changes in sleep patterns, or certain foods.

3. Inner ear infections: Infections in the inner ear, such as labyrinthitis or otitis media, can cause dizziness, nausea, and headache.

4. Meniere's disease: This is a disorder of the inner ear that can cause episodes of vertigo, nausea, and hearing loss.

5. Motion sickness: This can occur when the body receives conflicting signals from the eyes, inner ear, and other sensory systems. It can cause dizziness, nausea, and headache.

6. Anxiety or panic attacks: Anxiety or panic attacks can cause dizziness, nausea, and headache.

7. Low blood sugar (hypoglycemia): Low blood sugar can cause dizziness, headache, and nausea.

8. Medication side effects: Certain medications, such as antibiotics, antihistamines, and antidepressants, can cause dizziness, headache, and nausea as side effects.

9. Stroke or transient ischemic attack (TIA): These can cause sudden onset of dizziness, headache, and nausea.

10. Brain tumor or other central nervous system disorders: Rarely, a brain tumor or other central nervous system disorder can cause dizziness, headache, and nausea.

It is essential to consult a healthcare professional for a proper evaluation and diagnosis. They will likely perform a physical examination, take a detailed medical history, and order appropriate tests to determine the underlying cause of the symptoms. Treatment will depend on the specific diagnosis and may include medications, lifestyle changes, or other interventions.

高级用法

prompt = '''
A 52-year-old woman comes to the physician because of a 6-month history of generalized fatigue, low-grade fever, and a 10-kg (22-lb) weight loss. Physical examination shows generalized pallor and splenomegaly. Her hemoglobin concentration is 7.5 g/dL and leukocyte count is 41,800/mm3. Leukocyte alkaline phosphatase activity is low. Peripheral blood smear shows basophilia with myelocytes and metamyelocytes. Bone marrow biopsy shows cellular hyperplasia with proliferation of immature granulocytic cells. Which of the following mechanisms is most likely responsible for this patient's condition?
    Output JSON in this format.
    {
    "age":
    "gender"
    "past medicial history"
    "present symptoms"
    "interventions"
    "lab tests"
    "medications"
    "possible diseases"
    }
    
    PLEASE WRITE ONLY JSON
'''
response = assistant.generate_response(prompt)
print(response)

{
    "age": 52,
    "gender": "female",
    "past medical history": "unknown",
    "present symptoms": "generalized fatigue, low-grade fever, 10-kg (22-lb) weight loss",
    "interventions": "bone marrow biopsy",
    "lab tests": "hemoglobin concentration: 7.5 g/dL, leukocyte count: 41,800/mm3, leukocyte alkaline phosphatase activity: low, peripheral blood smear: basophilia with myelocytes and metamyelocytes",
    "medications": "none",
    "possible diseases": "chronic myeloid leukemia"
}

📚 详细文档

量化版本

本模型的量化版本可在 [mradermacher/Medichat - V2 - Llama3 - 8B - GGUF](https://huggingface.co/mradermacher/Medichat - V2 - Llama3 - 8B - GGUF) 找到。

📄 许可证

本模型使用的许可证为 other。