jokerbit
/

flux-qa-8

Model card Files Files and versions

jokerbit commited on Feb 13, 2025

Commit

0211f19

·

verified ·

1 Parent(s): 25e21d6

VAE again

Files changed (1) hide show

src/pipeline.py +1 -1

src/pipeline.py CHANGED Viewed

@@ -56,7 +56,7 @@ pipeline = FluxPipeline.from_pretrained(
 pipeline.transformer.to(memory_format=torch.channels_last)
 my_quantize(pipeline.vae)
-pipeline.vae = torch.compile(pipeline.vae, mode="max-autotune-no-cudagraphs")
 pipeline.to("cuda")
 def load_pipeline():

 pipeline.transformer.to(memory_format=torch.channels_last)
 my_quantize(pipeline.vae)
+pipeline.vae = torch.compile(pipeline.vae, mode="reduce-overhead")
 pipeline.to("cuda")
 def load_pipeline():