poipii
/

lpw_high_res_fix_pipeline

Model card Files Files and versions Community

poipiii commited on Apr 19, 2023

Commit

a86f18e

•

1 Parent(s): b248b08

fix unpacking of image

Files changed (1) hide show

pipeline.py +2 -1

pipeline.py CHANGED Viewed

@@ -1185,7 +1185,8 @@ class StableDiffusionLongPromptWeightingPipeline(StableDiffusionPipeline):
             callback_steps=callback_steps,
         )
-        resized_image = torch.nn.functional.interpolate(image, scale_factor=resize_scale)
         highres_image = self.__call__(prompt=prompt,
                                       negative_prompt=negative_prompt,

             callback_steps=callback_steps,
         )
+        resized_image = torch.nn.functional.interpolate(
+            image.images[0], scale_factor=resize_scale)
         highres_image = self.__call__(prompt=prompt,
                                       negative_prompt=negative_prompt,