Upload dc.py

dc.py

@@ -308,7 +308,6 @@ class GuiSD:
         self.model = None
     
         print("Loading model...")
-        self.model.device = torch.device("cpu")
         self.model = Model_Diffusers(
             base_model_id="cagliostrolab/animagine-xl-3.1",
             task_name="txt2img",
@@ -317,6 +316,7 @@ class GuiSD:
             retain_task_model_in_cache=False,
             #device="cpu",
         )
+        self.model.device = torch.device("cpu")
 
     def infer_short(self, model, pipe_params, progress=gr.Progress(track_tqdm=True)):
         progress(0, desc="Start inference...")