linoyts HF Staff commited on
Commit
7d0ff05
·
verified ·
1 Parent(s): 6dd10bd

pin torch version to fix CUBLAS error

Browse files
Files changed (1) hide show
  1. 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