Update gui.py

gui.py

@@ -28,9 +28,10 @@ class GuiSD:
         self.model = Model_Diffusers(
             base_model_id="eienmojiki/Anything-XL",
             task_name="txt2img",
-            vae_model="vaes/sdXL_v10VAEFix.safetensors",
+            vae_model=None,
             type_model_precision=torch.float16,
             retain_task_model_in_cache=False,
+            device="cpu",
         )
         self.model_list = model_list
         self.task_stablepy = task_stablepy
@@ -67,6 +68,8 @@ class GuiSD:
             if incompatible_vae:
                 vae_model = None
 
+        self.model.device = torch.device("cpu")
+        
         self.model.load_pipe(
             model_name,
             task_name=self.task_stablepy[task],
@@ -76,7 +79,7 @@ class GuiSD:
         )
         yield f"Model loaded: {model_name}"
 
-    @spaces.GPU(duration=80)
+    @spaces.GPU
     def generate_pipeline(
             self,
             prompt,
@@ -252,6 +255,7 @@ class GuiSD:
 
         # First load
         model_precision = torch.float16
+        self.model.device = torch.device("cuda:0")
         if not self.model:
             from modelstream import Model_Diffusers2