AlekseyCalvin commited on
Commit
9c77a26
1 Parent(s): 820759e

Update pipeline.py

Browse files
Files changed (1) hide show
  1. pipeline.py +1 -1
pipeline.py CHANGED
@@ -207,7 +207,7 @@ class FluxWithCFGPipeline(StableDiffusion3Pipeline):
207
  image = vae.decode(latents, return_dict=False)[0]
208
  return self.image_processor.postprocess(image, output_type=output_type)[0]
209
 
210
- class FluxWithCFGPipeline(FluxPipeline):
211
  @torch.inference_mode()
212
  def generate_image(
213
  self,
 
207
  image = vae.decode(latents, return_dict=False)[0]
208
  return self.image_processor.postprocess(image, output_type=output_type)[0]
209
 
210
+ class FluxWithCFGPipeline(StableDiffusion3Pipeline):
211
  @torch.inference_mode()
212
  def generate_image(
213
  self,