VLAlert / requirements.txt
AsianPlayer's picture
Add VLAlert code
1e05592 verified
Raw
History Blame Contribute Delete
489 Bytes
# requirements.txt
# 核心依赖
torch>=2.1.0
torchvision>=0.16.0
transformers>=4.46.0
accelerate>=0.34.0
peft>=0.13.0
# Qwen-VL特定
qwen-vl-utils
einops>=0.8.0
# 数据处理
opencv-python>=4.8.0
pillow>=10.0.0
numpy>=1.24.0
pandas>=2.0.0
scipy>=1.10.0
# 训练工具
wandb>=0.16.0
tensorboard>=2.15.0
tqdm>=4.66.0
# 配置管理
pyyaml>=6.0
omegaconf>=2.3.0
# 评估
scikit-learn>=1.3.0
matplotlib>=3.7.0
seaborn>=0.12.0
# 开发工具
pytest>=7.4.0
black>=23.0.0
flake8>=6.1.0