Update model.py

model.py

@@ -448,10 +448,11 @@ class StreamMultiDiffusion(nn.Module):
         negative_prompt: Optional[str] = None,
     ) -> bool:
         flag_changed = False
-        print('111111111111111111111111', self.device, self.pipe.device)
         if image is not None:
             image_ = image.resize((self.width, self.height))
+            print('111111111111111111111111', self.device, self.pipe.device)
             prompt = self.get_text_prompts(image_) if prompt is None else prompt
+            print('2222222222222222222222', self.device, self.pipe.device)
             negative_prompt = '' if negative_prompt is None else negative_prompt
             embed = self.pipe.encode_prompt(
                 prompt=[prompt],