diff --git a/environments-and-requirements/requirements-base.txt b/environments-and-requirements/requirements-base.txt index f125544f53..867273928e 100644 --- a/environments-and-requirements/requirements-base.txt +++ b/environments-and-requirements/requirements-base.txt @@ -31,7 +31,7 @@ taming-transformers-rom1504 test-tube>=0.7.5 torch-fidelity torchmetrics -transformers==4.21.* +transformers~=4.24 picklescan git+https://github.com/invoke-ai/GFPGAN@basicsr-1.4.1#egg=gfpgan ; platform_system == 'Windows' git+https://github.com/invoke-ai/GFPGAN@basicsr-1.4.2#egg=gfpgan ; platform_system != 'Windows' diff --git a/installer/requirements.in b/installer/requirements.in index 86f6be4434..8699ae8861 100644 --- a/installer/requirements.in +++ b/installer/requirements.in @@ -3,7 +3,8 @@ --trusted-host https://download.pytorch.org accelerate~=0.14 albumentations -diffusers +diffusers[torch]~=0.9 +einops eventlet flask_cors flask_socketio