diff --git a/requirements.txt b/requirements.txt index 8f488b3..18147bc 100644 --- a/requirements.txt +++ b/requirements.txt @@ -4,7 +4,7 @@ safetensors timm tomesd accelerate -torch +torch>=2.2.0 torchdiffeq torchsde xformers @@ -21,4 +21,4 @@ imageio[pyav] tensorboard gradio==3.41.2 diffusers==0.27.0 -transformers==4.37.2 \ No newline at end of file +transformers==4.37.2