John6666 commited on
Commit
9eb205f
·
verified ·
1 Parent(s): 9ee7707

Upload dc.py

Browse files
Files changed (1) hide show
  1. dc.py +1 -0
dc.py CHANGED
@@ -308,6 +308,7 @@ class GuiSD:
308
  self.model = None
309
 
310
  print("Loading model...")
 
311
  self.model = Model_Diffusers(
312
  base_model_id="cagliostrolab/animagine-xl-3.1",
313
  task_name="txt2img",
 
308
  self.model = None
309
 
310
  print("Loading model...")
311
+ self.model.device = torch.device("cpu")
312
  self.model = Model_Diffusers(
313
  base_model_id="cagliostrolab/animagine-xl-3.1",
314
  task_name="txt2img",