Spaces:
Running
Running
Update requirements.txt
Browse files- requirements.txt +3 -1
requirements.txt
CHANGED
|
@@ -1,3 +1,5 @@
|
|
|
|
|
|
|
|
| 1 |
torch>=2.3
|
| 2 |
torchvision>=0.18
|
| 3 |
diffusers>=0.29
|
|
@@ -7,7 +9,7 @@ einops>=0.8
|
|
| 7 |
omegaconf>=2.3
|
| 8 |
opencv-python>=4.10
|
| 9 |
matplotlib>=3.9
|
| 10 |
-
|
| 11 |
accelerate>=0.32
|
| 12 |
av>=12.2
|
| 13 |
opencv-python
|
|
|
|
| 1 |
+
--extra-index-url https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/onnxruntime-cuda-12/pypi/simple
|
| 2 |
+
onnxruntime-gpu
|
| 3 |
torch>=2.3
|
| 4 |
torchvision>=0.18
|
| 5 |
diffusers>=0.29
|
|
|
|
| 9 |
omegaconf>=2.3
|
| 10 |
opencv-python>=4.10
|
| 11 |
matplotlib>=3.9
|
| 12 |
+
nvidia-cudnn-cu12==9.3.0.75
|
| 13 |
accelerate>=0.32
|
| 14 |
av>=12.2
|
| 15 |
opencv-python
|