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('cuda', dtype)
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)