diff --git a/requirements.txt b/requirements.txt index e3077e2..6b32a72 100644 --- a/requirements.txt +++ b/requirements.txt @@ -10,5 +10,5 @@ pytest yacs markupsafe==2.0.1 scikit-image==0.19.3 -diffusers==0.6.0 +diffusers[torch]==0.7.2 transformers==4.21.0