Flux-Use-Compiled-Graph

Running on Zero

sayakpaul HF Staff commited on Sep 11

Commit

f7ab8c7

verified ·

1 Parent(s): 3b97764

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -15,7 +15,7 @@ pipeline = FluxPipeline.from_pretrained(
     "black-forest-labs/Flux.1-Dev", torch_dtype=torch.bfloat16
 ).to(device)
 pipeline.transformer.set_attn_processor(FlashFluxAttnProcessor3_0())
-aoti_load_(pipeline.transformer, "sayakpaul/flux-dev-aot", "flux-dev-aot.pt2")
 @spaces.GPU

     "black-forest-labs/Flux.1-Dev", torch_dtype=torch.bfloat16
 ).to(device)
 pipeline.transformer.set_attn_processor(FlashFluxAttnProcessor3_0())
+aoti_load_(pipeline.transformer, "zerogpu-aoti/flux-dev-aot", "flux-dev-aot.pt2")
 @spaces.GPU