poipii
/

lpw_high_res_fix_pipeline

Model card Files Files and versions Community

poipiii commited on Apr 19, 2023

Commit

b248b08

·

1 Parent(s): 4f9614f

fix interpolation mode

Files changed (1) hide show

pipeline.py +1 -1

pipeline.py CHANGED Viewed

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

             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,