diff --git a/requirements.txt b/requirements.txt index dfd2d67..54d3976 100644 --- a/requirements.txt +++ b/requirements.txt @@ -2,7 +2,8 @@ diffusers>=0.30.1 #git+https://github.com/huggingface/diffusers.git@main#egg=dif transformers>=4.44.0 # The development team is working on version 0.44.2 accelerate>=0.33.0 #git+https://github.com/huggingface/accelerate.git@main#egg=accelerate is suggested sentencepiece>=0.2.0 # T5 used -SwissArmyTransformer>=0.4.12 # Inference +SwissArmyTransformer>=0.4.12 +numpy==1.26.0 torch>=2.4.0 # Tested in 2.2 2.3 2.4 and 2.5, The development team is working on version 2.4.0. torchvision>=0.19.0 # The development team is working on version 0.19.0. gradio>=4.42.0 # For HF gradio demo @@ -11,4 +12,4 @@ imageio==2.34.2 # For diffusers inference export video imageio-ffmpeg==0.5.1 # For diffusers inference export video openai>=1.42.0 # For prompt refiner moviepy==1.0.3 # For export video -pillow==9.5.0 # For moviepy \ No newline at end of file +pillow==9.5.0 \ No newline at end of file