Spaces:
Runtime error
Runtime error
Update model.py
Browse files
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
|