# Fusion 项目依赖

# 核心依赖
torch>=2.0.0,<3.0.0
transformers>=4.36.0,<5.0.0
accelerate>=0.25.0,<1.0.0
datasets>=2.16.0,<3.0.0

# PEFT (LoRA/QLoRA)
peft>=0.7.0,<1.0.0
bitsandbytes>=0.41.0,<1.0.0

# 分布式训练
deepspeed>=0.12.0,<0.16.0

# 数据处理
numpy>=1.24.0,<2.0.0
pandas>=2.0.0,<3.0.0
tqdm>=4.66.0,<5.0.0

# Tokenizer
sentencepiece>=0.1.99,<0.3.0
tokenizers>=0.15.0,<1.0.0

# 语言识别（数据清洗）
langid>=1.1.6,<2.0.0

# 推理部署
ollama>=0.1.0,<1.0.0
# 注意：Ollama 也需要从 https://ollama.com/ 下载二进制客户端

# 量化工具依赖
scipy>=1.11.0,<2.0.0
scikit-learn>=1.3.0,<2.0.0

# 评估（可选）
evaluate>=0.4.0,<1.0.0
rouge-score>=0.1.2,<1.0.0

# 可视化（可选）
tensorboard>=2.15.0,<3.0.0
wandb>=0.16.0,<1.0.0
gradio>=4.0.0,<6.0.0  # inference/dashboard.py 依赖

# ONNX 导出
onnx>=1.14.0,<2.0.0
onnxruntime>=1.16.0,<2.0.0

# 工具
pyyaml>=6.0.0,<7.0.0
click>=8.1.0,<9.0.0

# 测试
pytest>=7.4.0,<9.0.0
pytest-cov>=4.1.0,<6.0.0