Update inferencer.py

inferencer.py

@@ -315,8 +315,15 @@ class UniPicV2Inferencer:
         embeds = self._process_inputs(inputs, num_queries)
         
         generator = torch.Generator(device=self.device).manual_seed(seed)
+
+        latents = torch.randn(
+            1, self.pipeline.transformer.config.in_channels,
+            height // self.pipeline.vae_scale_factor,
+            width // self.pipeline.vae_scale_factor,
+            device=self.device, dtype=self.pipeline.transformer.dtype)
         
         edited_image = self.pipeline(
+            latents=latents,
             image=image,
             prompt_embeds=embeds["prompt_embeds"].to(self.device),
             pooled_prompt_embeds=embeds["pooled_prompt_embeds"].to(self.device),