Spaces:
Running
on
Zero
Running
on
Zero
AlekseyCalvin
commited on
Commit
•
9c77a26
1
Parent(s):
820759e
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,
|