poipii
/

lpw_high_res_fix_pipeline

Model card Files Files and versions Community

poipiii commited on Apr 19, 2023

Commit

9bee773

•

1 Parent(s): a86f18e

fix scale size

Files changed (1) hide show

pipeline.py +1 -1

pipeline.py CHANGED Viewed

@@ -1186,7 +1186,7 @@ class StableDiffusionLongPromptWeightingPipeline(StableDiffusionPipeline):
         )
         resized_image = torch.nn.functional.interpolate(
-            image.images[0], scale_factor=resize_scale)
         highres_image = self.__call__(prompt=prompt,
                                       negative_prompt=negative_prompt,

         )
         resized_image = torch.nn.functional.interpolate(
+            image.images[0],size = (int(height*resize_scale), int(width*resize_scale)))
         highres_image = self.__call__(prompt=prompt,
                                       negative_prompt=negative_prompt,