Commit ·
072c816
1
Parent(s): 6dd10bd
pin torch version to fix CUBLAS error (#34)
Browse files- pin torch version to fix CUBLAS error (7d0ff05e62d749c93a9032628f7bc5e034f7e644)
Co-authored-by: Linoy Tsaban <linoyts@users.noreply.huggingface.co>
- requirements.txt +2 -1
requirements.txt
CHANGED
|
@@ -12,4 +12,5 @@ imageio-ffmpeg
|
|
| 12 |
einops
|
| 13 |
timm
|
| 14 |
av
|
| 15 |
-
git+https://github.com/huggingface/diffusers.git@main
|
|
|
|
|
|
| 12 |
einops
|
| 13 |
timm
|
| 14 |
av
|
| 15 |
+
git+https://github.com/huggingface/diffusers.git@main
|
| 16 |
+
torch==2.9.1
|