| # DreamZero 依赖清单 | |
| # 安装: pip install -r requirements.txt | |
| # 或完整安装: pip install -e .[train] | |
| # 核心深度学习框架 | |
| torch>=2.3,<2.7 | |
| torchvision>=0.18 | |
| transformers==4.51.3 | |
| diffusers>=0.30,<0.32 | |
| accelerate>=0.27 | |
| # 分布式训练 | |
| deepspeed>=0.12,<0.16 | |
| # 配置 | |
| hydra-core>=1.3 | |
| omegaconf>=2.3 | |
| # 数据/视频处理 | |
| decord | |
| opencv-python>=4.8 | |
| av | |
| einops>=0.8 | |
| # 序列化 | |
| safetensors | |
| sentencepiece | |
| protobuf | |
| # 数据格式 | |
| zarr | |
| h5py | |
| pyarrow | |
| pandas | |
| scipy | |
| # 数值 | |
| numpy<2 | |
| tqdm | |
| # 显存优化 | |
| bitsandbytes>=0.41 | |
| # 实验追踪 | |
| wandb | |
| tensorboard | |