Update custom_pipeline.py

custom_pipeline.py

@@ -48,7 +48,7 @@ class FluxWithCFGPipeline(FluxPipeline):
     with progressively increasing resolution for faster generation.
     """
     @torch.inference_mode()
-    async def generate_images(
+    def generate_images(
         self,
         prompt: Union[str, List[str]] = None,
         prompt_2: Optional[Union[str, List[str]]] = None,
@@ -138,7 +138,7 @@ class FluxWithCFGPipeline(FluxPipeline):
             timestep = t.expand(latents.shape[0]).to(latents.dtype)
 
             if generate_with_graph:
-                return await generate_with_graph(latents, prompt_embeds, pooled_prompt_embeds, text_ids, latent_image_ids, timestep)
+                return generate_with_graph(latents, prompt_embeds, pooled_prompt_embeds, text_ids, latent_image_ids, timestep)
             else:
                 noise_pred = self.transformer(
                     hidden_states=latents,