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",
|