diff --git a/requirements.txt b/requirements.txt index ee1b080..b268439 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,6 +1,7 @@ -torch==2.0.1 --index-url https://download.pytorch.org/whl/cu118 -torchvision==0.15.2 --index-url https://download.pytorch.org/whl/cu118 -torchaudio==2.0.2 --index-url https://download.pytorch.org/whl/cu118 +--extra-index-url https://download.pytorch.org/whl/cu118 +torch==2.0.1 +torchvision==0.15.2 +torchaudio==2.0.2 diffusers==0.27.2 accelerate==0.28.0 tensorflow==2.12.0 @@ -12,3 +13,9 @@ transformers==4.39.2 gdown requests imageio[ffmpeg] + +omegaconf +ffmpeg-python +gradio +spaces +moviepy