requirements.txt

# 核心框架
torch==2.4.0                   # 保持与conda环境一致（pypi安装）
torchvision==0.19.0            # 保持与conda环境一致（pypi安装）

# Hugging Face 生态
transformers==4.44.0           # 原4.39.1 → conda实际安装4.44.0
diffusers==0.31.0              # 保持与conda环境一致
accelerate==0.34.2             # 原0.27.2 → conda实际安装0.34.2
peft==0.12.0                   # 原git提交 → conda实际安装0.12.0（pypi）
datasets==2.20.0               # 保持与conda环境一致

# 工具链
deepspeed==0.15.4              # 原0.14.2 → conda实际安装0.15.4
# bitsandbytes==0.44.1           # 原0.43.0 → conda实际安装0.44.1
protobuf==3.20.0               # 原3.20.3 → conda实际安装3.20.0（需验证tensorboard兼容性）
tensorboard==2.18.0            # 新增明确版本（conda实际安装2.18.0）
tensorboardx==2.6.2.2          # 新增明确版本（conda实际安装2.6.2.2）
webdataset==0.2.100            # 新增明确版本（conda实际安装0.2.100）

# 训练辅助
warmup_scheduler==0.3          # 新增明确版本（conda实际安装0.3）
torchmetrics==1.6.0            # 新增明确版本（conda实际安装1.6.0）
open_clip_torch==2.29.0        # 新增明确版本（conda实际安装2.29.0）
evaluate==0.4.3                # 新增明确版本（conda实际安装0.4.3）
bert_score==0.3.13             # 新增明确版本（conda实际安装0.3.13）
einops==0.8.0                  # 保持与conda环境一致
wandb==0.17.7                  # 保持与conda环境一致

# 图像处理
matplotlib==3.9.2              # 新增明确版本（conda实际安装3.9.2）
opencv-python==4.10.0.84       # 新增明确版本（conda实际安装4.10.0.84）
clean-fid==0.1.35              # 新增明确版本（conda实际安装0.1.35）
skia-python==87.6              # 新增明确版本（conda实际安装87.6）

# 部署与接口
# gradio==5.5.0                  # 新增明确版本（conda实际安装5.5.0）
langchain>=0.0.139              # 保持约束（conda实际安装0.3.7符合要求）
tiktoken==0.8.0                # 新增明确版本（conda实际安装0.8.0）

# 系统工具
ninja==1.11.1.1                # 新增明确版本（conda实际安装1.11.1.1）
pynvml==11.5.3                 # 新增明确版本（conda实际安装11.5.3）
colorama==0.4.6                # 新增明确版本（conda实际安装0.4.6）
click>=8.0.4,<9                # 保持约束（conda实际安装8.1.7符合要求）\

sentencepiece
# random