| --extra-index-url https://download.pytorch.org/whl/cu121 |
|
|
| # cu121 is to meet our nvidia env. you can change it to other versions |
| torch==2.2.0+cu121 |
| torchvision==0.17.0+cu121 |
|
|
| # Core dependencies for LDMAE |
| timm==1.0.12 |
| diffusers==0.32.1 |
| accelerate |
| torchdiffeq |
| pytorch_fid |
| tensorboard==2.16.2 |
| omegaconf==2.3.0 |
| einops |
| fairscale |
| safetensors |
| taming-transformers |
|
|
| # Additional dependencies for data processing |
| Pillow>=9.0.0 |
| tqdm |
| numpy<2.0 |
| pyyaml |
|
|
| transformers |
| scipy |
| matplotlib |
| opencv-python<4.9.0 |
| scikit-learn |
| lpips |
|
|
| # Optional: for reconstruction evaluation |
| # torchmetrics |
|
|
| # Optional: for LPIPS loss (if using taming-transformers) |
| # taming-transformers-rom1504 |
