From 9cae83066e034a1abd83520391acdd9d8580aee4 Mon Sep 17 00:00:00 2001 From: yunkchen Date: Fri, 31 May 2024 11:19:14 +0800 Subject: [PATCH] Update requirements.txt Add torch version limit --- requirements.txt | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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