LibreFLUX_LoRAs_Gallery

Runtime error

AlekseyCalvin commited on Oct 14, 2024

Commit

fcfc6b2

verified ·

1 Parent(s): 7befb0b

Update custom_pipeline.py

Files changed (1) hide show

custom_pipeline.py CHANGED Viewed

@@ -58,7 +58,7 @@ def prepare_timesteps(
 class FluxWithCFGPipeline(FluxPipeline):
     @torch.inference_mode()
-    def generate_images(
         self,
         prompt: Union[str, List[str]] = None,
         prompt_2: Optional[Union[str, List[str]]] = None,

 class FluxWithCFGPipeline(FluxPipeline):
     @torch.inference_mode()
+    def generate_image(
         self,
         prompt: Union[str, List[str]] = None,
         prompt_2: Optional[Union[str, List[str]]] = None,