Spaces:
Runtime error
Runtime error
Update pipeline.py
Browse files- 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(
|
| 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,
|