LibreFLUX_LoRAs_Gallery

Runtime error

AlekseyCalvin commited on Oct 15, 2024

Commit

bf79b8a

verified ·

1 Parent(s): 2bcefbc

Update pipeline.py

Files changed (1) hide show

pipeline.py CHANGED Viewed

@@ -548,7 +548,8 @@ class FluxWithCFGPipeline(DiffusionPipeline, FluxLoraLoaderMixin, FromSingleFile
         return self._interrupt
     @torch.no_grad()
-    def __call__(
         self,
         prompt: Union[str, List[str]] = None,
         prompt_2: Optional[Union[str, List[str]]] = None,

         return self._interrupt
     @torch.no_grad()
+    @torch.inference_mode()
+    def generate_image(
         self,
         prompt: Union[str, List[str]] = None,
         prompt_2: Optional[Union[str, List[str]]] = None,