| # Use PyTorch CUDA 12.1 wheels for torch/torchvision | |
| --index-url https://download.pytorch.org/whl/cu121 | |
| torch==2.3.1 | |
| torchvision==0.18.1 | |
| # Core utils | |
| pyyaml>=6.0.1 | |
| tqdm>=4.66.0 | |
| numpy>=1.26.0 | |
| Pillow>=10.0.0 | |
| einops>=0.7.0 | |
| # Logging & videos | |
| wandb>=0.16.0 | |
| imageio>=2.31.0 | |
| imageio-ffmpeg>=0.4.9 # để ghi MP4 mà không cần ffmpeg hệ thống | |
| # Metrics (nếu bật FID/IS) | |
| clean-fid>=0.1.35 | |
| torch-fidelity>=0.3.0 | |
| # (Tùy chọn) Đẩy model lên Hugging Face | |
| huggingface_hub>=0.23.0 | |