diff --git a/environment.yaml b/environment.yaml index f36b0e17d..22e10c39b 100644 --- a/environment.yaml +++ b/environment.yaml @@ -21,7 +21,9 @@ dependencies: - streamlit>=0.73.1 - einops==0.3.0 - torch-fidelity==0.3.0 - - transformers==4.3.1 + - transformers==4.6.0 + - torchmetrics<0.6.0 - -e git+https://github.com/CompVis/taming-transformers.git@master#egg=taming-transformers - -e git+https://github.com/openai/CLIP.git@main#egg=clip - - -e . \ No newline at end of file + - -e . + - packaging==21.3