Update requirements.txt
Browse files- requirements.txt +6 -3
requirements.txt
CHANGED
@@ -1,11 +1,14 @@
|
|
1 |
-
torch==2.
|
2 |
-
|
3 |
-
|
|
|
4 |
accelerate
|
5 |
transformers
|
6 |
scipy
|
7 |
opencv-python
|
8 |
omegaconf
|
9 |
einops
|
|
|
10 |
roma
|
11 |
jaxtyping
|
|
|
|
1 |
+
torch==2.5.1+cu118
|
2 |
+
torchaudio==2.4.1
|
3 |
+
torchvision==0.19.1
|
4 |
+
diffusers==0.31.0
|
5 |
accelerate
|
6 |
transformers
|
7 |
scipy
|
8 |
opencv-python
|
9 |
omegaconf
|
10 |
einops
|
11 |
+
git+https://github.com/pesser/splatting
|
12 |
roma
|
13 |
jaxtyping
|
14 |
+
numpy<2 # Downgrade NumPy to avoid compatibility issues
|