AlekseyCalvin commited on
Commit
acc9fdc
·
verified ·
1 Parent(s): 08a702c

Update pipeline.py

Browse files
Files changed (1) hide show
  1. pipeline.py +0 -3
pipeline.py CHANGED
@@ -499,8 +499,6 @@ class FluxWithCFGPipeline(DiffusionPipeline, FluxLoraLoaderMixin, FromSingleFile
499
  cfg: Optional[bool] = True,
500
  return_dict: bool = True,
501
  joint_attention_kwargs: Optional[Dict[str, Any]] = None,
502
- callback_on_step_end: Optional[Callable[[int, int, Dict], None]] = None,
503
- callback_on_step_end_tensor_inputs: List[str] = ["latents"],
504
  max_sequence_length: int = 512,
505
  ):
506
  height = height or self.default_sample_size * self.vae_scale_factor
@@ -514,7 +512,6 @@ class FluxWithCFGPipeline(DiffusionPipeline, FluxLoraLoaderMixin, FromSingleFile
514
  width,
515
  prompt_embeds=prompt_embeds,
516
  pooled_prompt_embeds=pooled_prompt_embeds,
517
- callback_on_step_end_tensor_inputs=callback_on_step_end_tensor_inputs,
518
  max_sequence_length=max_sequence_length,
519
  )
520
 
 
499
  cfg: Optional[bool] = True,
500
  return_dict: bool = True,
501
  joint_attention_kwargs: Optional[Dict[str, Any]] = None,
 
 
502
  max_sequence_length: int = 512,
503
  ):
504
  height = height or self.default_sample_size * self.vae_scale_factor
 
512
  width,
513
  prompt_embeds=prompt_embeds,
514
  pooled_prompt_embeds=pooled_prompt_embeds,
 
515
  max_sequence_length=max_sequence_length,
516
  )
517