Spaces:
Running
on
Zero
Running
on
Zero
Upload dc.py
Browse files
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",
|