Upload folder using huggingface_hub

app.py

@@ -59,7 +59,7 @@ def infer(
         num_images_per_caption=2,
         positive_prompt=positive_prompt,
         negative_prompt=negative_prompt,
-        hw=(512, 512),
+        hw=(input_image.size[1], input_image.size[0]),
         cfg=text_cfg,
         cfg_img=img_cfg,
         cfg_schedule="constant",

models/gen_pipeline.py

@@ -387,7 +387,7 @@ class NextStepPipeline:
         )
 
         # 7. unpatchify
-        latents = self.model.unpatchify(tokens)
+        latents = self.model.unpatchify(tokens, h=hw[0] // self.down_factor, w=hw[1] // self.down_factor)
         latents = (latents / self.scaling_factor) + self.shift_factor
 
         # 8. decode latents