Spaces:
Running on Zero
Running on Zero
Update pipeline_stable_diffusion_xl_instantid.py
#19
by DaSilvaRage - opened
pipeline_stable_diffusion_xl_instantid.py
CHANGED
|
@@ -486,7 +486,7 @@ class LongPromptWeight(object):
|
|
| 486 |
class StableDiffusionXLInstantIDPipeline(StableDiffusionXLControlNetPipeline):
|
| 487 |
|
| 488 |
def cuda(self, dtype=torch.float16, use_xformers=False):
|
| 489 |
-
self.to('
|
| 490 |
|
| 491 |
if hasattr(self, 'image_proj_model'):
|
| 492 |
self.image_proj_model.to(self.unet.device).to(self.unet.dtype)
|
|
|
|
| 486 |
class StableDiffusionXLInstantIDPipeline(StableDiffusionXLControlNetPipeline):
|
| 487 |
|
| 488 |
def cuda(self, dtype=torch.float16, use_xformers=False):
|
| 489 |
+
self.to('cpu', dtype)
|
| 490 |
|
| 491 |
if hasattr(self, 'image_proj_model'):
|
| 492 |
self.image_proj_model.to(self.unet.device).to(self.unet.dtype)
|