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
|