这是一个经过训练的智慧中医诊断助手模型, 基于DeepSeek-R1-Distill-Llama-8B微调而来. 该模型可以帮助医生和患者进行中医诊断和咨询.
1from transformers import AutoTokenizer, AutoModelForCausalLM
2
3# 加载模型和分词器
4tokenizer = AutoTokenizer.from_pretrained("yyxa9/zhongyi315-3")
5model = AutoModelForCausalLM.from_pretrained("yyxa9/zhongyi315-3")
6
7# 使用示例
8prompt = """以下是描述任务的指令，以及提供进一步上下文的输入。
9请写出一个适当完成请求的回答。
10在回答之前，请仔细思考问题，并创建一个逻辑连贯的思考过程，以确保回答准确无误。
11
12### 指令：
13你是一位经验丰富的中医师，擅长运用中医理论分析病案，进行辨证论治。
14请根据患者的症状、体征和舌脉等信息，进行中医诊断分析并提供合理的治疗方案。
15
16### 问题：
17患者李某，男，45岁，主诉头痛、眩晕3个月，伴有耳鸣、健忘，舌质红，苔薄黄，脉弦细。"""
18
19inputs = tokenizer(prompt, return_tensors="pt")
20outputs = model.generate(**inputs, max_length=2048, temperature=0.7, top_p=0.9)
21print(tokenizer.decode(outputs[0], skip_special_tokens=True))
