reset requirements

model.py

@@ -61,8 +61,7 @@ class Model:
         if self.device.type == "cuda":
             pipe.disable_xformers_memory_efficient_attention()
         pipe.to(self.device)
-        pipe.enable_model_cpu_offload()
-        pipe.enable_vae_slicing()
+
         torch.cuda.empty_cache()
         gc.collect()
         self.base_model_id = base_model_id

requirements.txt

@@ -1,5 +1,6 @@
 controlnet-aux
-torch
+torch==2.0.1
+torchvision==0.15.2
 diffusers
 transformers
 safetensors
@@ -7,5 +8,5 @@ accelerate
 kornia
 huggingface_hub
 gradio==4.26.0
-xformers
 mediapipe
+spaces