HuatuoGPT：中文医疗问诊大模型

python from transformers import AutoModelForCausalLM, AutoTokenizer

model_name = "FreedomIntelligence/HuatuoGPT-7B" tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True) model = AutoModelForCausalLM.from_pretrained(model_name, device_map="auto", trust_remote_code=True)

prompt = "用户：最近总是咳嗽，有痰但不多，喉咙痒。" response, history = model.chat(tokenizer, prompt, history=[]) print(response)