Spaces:
Runtime error
Runtime error
foz
commited on
Commit
·
046b08b
1
Parent(s):
f687279
Further cleaning - convert to jax
Browse files- requirements.txt +5 -0
requirements.txt
CHANGED
|
@@ -1,4 +1,9 @@
|
|
| 1 |
--extra-index-url https://download.pytorch.org/whl/cu113
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 2 |
torch
|
| 3 |
accelerate
|
| 4 |
decord==0.6.0
|
|
|
|
| 1 |
--extra-index-url https://download.pytorch.org/whl/cu113
|
| 2 |
+
flax
|
| 3 |
+
jax[cuda11_pip]
|
| 4 |
+
-f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
|
| 5 |
+
jaxlib
|
| 6 |
+
git+https://github.com/huggingface/diffusers@main
|
| 7 |
torch
|
| 8 |
accelerate
|
| 9 |
decord==0.6.0
|