mazpie commited on
Commit
1503429
1 Parent(s): cda2bf8

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +2 -0
requirements.txt CHANGED
@@ -30,4 +30,6 @@ transformers==4.31.0
30
  hydra-submitit-launcher==1.1.5
31
  moviepy==1.0.3
32
  imageio[ffmpeg]
 
 
33
  # for the demo: gradio==4.36.1
 
30
  hydra-submitit-launcher==1.1.5
31
  moviepy==1.0.3
32
  imageio[ffmpeg]
33
+ torch==2.2.0
34
+ torchvision
35
  # for the demo: gradio==4.36.1