Baize-TCM-LLM-V1-4bit 是一个专为中医药领域设计的大语言模型，基于通用大模型在 4,735 条高质量中医问答语料 上进行指令微调（Supervised Fine-Tuning）后，进一步采用 4bit 量化 技术压缩，显著降低显存占用，支持在消费级 GPU（如 16GB 显存）上高效推理。

该模型适用于：

中医智能问答系统
中医药知识辅助诊断
医学生教学与考试辅助
中医古籍解析与解释
低资源环境下的本地化部署

🏗️ 模型架构

项目	说明
基础模型	Qwen3-8B
微调数据	Baize-TCM-Corpus-for-Large-Language-Models-V1
参数量	~8B
量化方式	4bit
量化工具	`transformers`
上下文长度	2048 tokens
训练方式	LoRA

🚀 快速使用（Inference）

方法：使用 `transformers` 加载（推荐）

python

from transformers import AutoTokenizer, AutoModelForCausalLM, BitsAndBytesConfig
import torch

# 4bit 配置
bnb_config = BitsAndBytesConfig(
    load_in_4bit=True,
    bnb_4bit_quant_type="nf4",
    bnb_4bit_compute_dtype=torch.float16,
    bnb_4bit_use_double_quant=True,
)

model_name = "your-username/Baize-Traditional-Chinese-Medicine-Large-Language-Model-V1-4bit"

tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)
model = AutoModelForCausalLM.from_pretrained(
    model_name,
    quantization_config=bnb_config,
    device_map="auto",
    trust_remote_code=True
)

# 推理
prompt = "问题：气虚的主要症状有哪些？\n答案："
inputs = tokenizer(prompt, return_tensors="pt").to("cuda")

outputs = model.generate(
    **inputs,
    max_new_tokens=200,
    temperature=0.7,
    do_sample=True
)

print(tokenizer.decode(outputs[0], skip_special_tokens=True))

DigitalIntelligenceCenter-of-ICMM/Baize-Traditional-Chinese-Medicine-Large-Language-Model-V1-4bit

白泽中医药大语言模型 V1 - 4bit 量化版

📚 模型简介

🏗️ 模型架构

🚀 快速使用（Inference）

方法 ：使用 transformers 加载（推荐）

DigitalIntelligenceCenter-of-ICMM/
Baize-Traditional-Chinese-Medicine-Large-Language-Model-V1-4bit

方法：使用 `transformers` 加载（推荐）