Spaces:

zerogpu-aoti
/

FLUX.1-dev-fa3-aoti

Runtime error

cbensimon HF Staff commited on Jul 3

Commit

8b98990

1 Parent(s): f742b99

Useless here

Files changed (1) hide show

app.py CHANGED Viewed

@@ -25,9 +25,6 @@ pipeline = FluxPipeline.from_pretrained('black-forest-labs/FLUX.1-schnell', torc
 def compile_transformer():
     pipeline.transformer.fuse_qkv_projections()
-    pipeline.vae.fuse_qkv_projections()
-    pipeline.vae.to(memory_format=torch.channels_last)
     quantize_(pipeline.transformer, Float8DynamicActivationFloat8WeightConfig())
     def _example_tensor(*shape):

 def compile_transformer():
     pipeline.transformer.fuse_qkv_projections()
     quantize_(pipeline.transformer, Float8DynamicActivationFloat8WeightConfig())
     def _example_tensor(*shape):