AlekseyCalvin commited on
Commit
eb7660d
·
verified ·
1 Parent(s): fcfc6b2

Update pipeline.py

Browse files
Files changed (1) hide show
  1. pipeline.py +1 -1
pipeline.py CHANGED
@@ -209,7 +209,7 @@ class FluxWithCFGPipeline(FluxPipeline):
209
 
210
  class FluxWithCFGPipeline(FluxPipeline):
211
  @torch.inference_mode()
212
- def generate_images(
213
  self,
214
  prompt: Union[str, List[str]] = None,
215
  prompt_2: Optional[Union[str, List[str]]] = None,
 
209
 
210
  class FluxWithCFGPipeline(FluxPipeline):
211
  @torch.inference_mode()
212
+ def generate_image(
213
  self,
214
  prompt: Union[str, List[str]] = None,
215
  prompt_2: Optional[Union[str, List[str]]] = None,