diff --git a/environments-and-requirements/requirements-base.txt b/environments-and-requirements/requirements-base.txt new file mode 100644 index 0000000000..fde1c8e234 --- /dev/null +++ b/environments-and-requirements/requirements-base.txt @@ -0,0 +1,38 @@ +--prefer-binary + +# pip will resolve the version which matches torch +albumentations +dependency_injector==4.40.0 +diffusers +einops +eventlet +flask==2.1.3 +flask_cors==3.0.10 +flask_socketio==5.3.0 +flaskwebgui==0.3.7 +getpass_asterisk +gfpgan +huggingface-hub +imageio +imageio-ffmpeg +kornia +numpy +omegaconf +opencv-python +pillow +pip>=22 +pudb +pyreadline3 +pytorch-lightning==1.7.7 +realesrgan +scikit-image>=0.19 +send2trash +streamlit +taming-transformers-rom1504 +test-tube +torch-fidelity +torchmetrics +transformers==4.21.* +git+https://github.com/openai/CLIP.git@main#egg=clip +git+https://github.com/Birch-san/k-diffusion.git@mps#egg=k-diffusion +git+https://github.com/invoke-ai/clipseg.git@models-rename#egg=clipseg diff --git a/environments-and-requirements/requirements-lin-amd.txt b/environments-and-requirements/requirements-lin-amd.txt index 88c5e8b9f1..8ca0ead88c 100644 --- a/environments-and-requirements/requirements-lin-amd.txt +++ b/environments-and-requirements/requirements-lin-amd.txt @@ -1,4 +1,4 @@ --r environments-and-requirements/requirements.txt +-r environments-and-requirements/requirements-base.txt # Get hardware-appropriate torch/torchvision --extra-index-url https://download.pytorch.org/whl/rocm5.1.1 --trusted-host https://download.pytorch.org diff --git a/environments-and-requirements/requirements-lin-arm64.txt b/environments-and-requirements/requirements-lin-arm64.txt index 8c1fe4fa0e..304432e4c7 100644 --- a/environments-and-requirements/requirements-lin-arm64.txt +++ b/environments-and-requirements/requirements-lin-arm64.txt @@ -1,3 +1,3 @@ --pre torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/nightly/cpu --r environments-and-requirements/requirements.txt +-r environments-and-requirements/requirements-base.txt -e . diff --git a/environments-and-requirements/requirements-lin-cuda.txt b/environments-and-requirements/requirements-lin-cuda.txt index 24c309f99c..1578c60334 100644 --- a/environments-and-requirements/requirements-lin-cuda.txt +++ b/environments-and-requirements/requirements-lin-cuda.txt @@ -1,2 +1,2 @@ --r environments-and-requirements/requirements.txt +-r environments-and-requirements/requirements-base.txt -e . diff --git a/environments-and-requirements/requirements-mac-mps-cpu.txt b/environments-and-requirements/requirements-mac-mps-cpu.txt index 0f5b3ecbbe..05c7d9a77c 100644 --- a/environments-and-requirements/requirements-mac-mps-cpu.txt +++ b/environments-and-requirements/requirements-mac-mps-cpu.txt @@ -1,4 +1,4 @@ --r environments-and-requirements/requirements.txt +-r environments-and-requirements/requirements-base.txt protobuf==3.19.6 torch<1.13.0 diff --git a/environments-and-requirements/requirements-win-colab-cuda.txt b/environments-and-requirements/requirements-win-colab-cuda.txt index d14b0b1b1f..68ee1bace6 100644 --- a/environments-and-requirements/requirements-win-colab-cuda.txt +++ b/environments-and-requirements/requirements-win-colab-cuda.txt @@ -1,4 +1,4 @@ --r environments-and-requirements/requirements.txt +-r environments-and-requirements/requirements-base.txt # Get hardware-appropriate torch/torchvision --extra-index-url https://download.pytorch.org/whl/cu116 --trusted-host https://download.pytorch.org