Spaces:
Running
on
Zero
Running
on
Zero
File size: 478 Bytes
2f4febc 4f339f5 2f4febc |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 |
--find-links https://download.pytorch.org/whl/torch_stable.html accelerate>=0.25.0 torch==2.1.2 torchvision==0.16.2 transformers>=4.30.0 numpy>=1.23.5 kornia>=0.7.0 insightface>=0.7.3 opencv-python>=4.8.1.78 tqdm>=4.66.1 matplotlib>=3.7.4 webdataset>=0.2.79 wandb>=0.16.2 munch>=4.0.0 onnxruntime>=1.16.3 einops>=0.7.0 onnx2torch>=1.5.13 warmup-scheduler @ git+https://github.com/ildoonet/pytorch-gradual-warmup-lr.git torchtools @ git+https://github.com/pabloppp/pytorch-tools |