ironjr commited on
Commit
f99e693
·
verified ·
1 Parent(s): aece490

Update model.py

Browse files
Files changed (1) hide show
  1. model.py +1 -0
model.py CHANGED
@@ -448,6 +448,7 @@ class StreamMultiDiffusion(nn.Module):
448
  negative_prompt: Optional[str] = None,
449
  ) -> bool:
450
  flag_changed = False
 
451
  if image is not None:
452
  image_ = image.resize((self.width, self.height))
453
  prompt = self.get_text_prompts(image_) if prompt is None else prompt
 
448
  negative_prompt: Optional[str] = None,
449
  ) -> bool:
450
  flag_changed = False
451
+ print('111111111111111111111111', self.device, self.pipe.device)
452
  if image is not None:
453
  image_ = image.resize((self.width, self.height))
454
  prompt = self.get_text_prompts(image_) if prompt is None else prompt