Skip to content

Commit b00a564

Browse files
authored
[docs] add note about use_duck_shape in auraflow docs. (#11348)
add note about use_duck_shape in auraflow docs.
1 parent efc9d68 commit b00a564

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

docs/source/en/api/pipelines/aura_flow.md

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -100,6 +100,8 @@ AuraFlow can be compiled with `torch.compile()` to speed up inference latency ev
100100
)
101101
```
102102

103+
Specifying `use_duck_shape` to be `False` instructs the compiler if it should use the same symbolic variable to represent input sizes that are the same. For more details, check out [this comment](https://github.com/huggingface/diffusers/pull/11327#discussion_r2047659790).
104+
103105
This enables from 100% (on low resolutions) to a 30% (on 1536x1536 resolution) speed improvements.
104106

105107
Thanks to [AstraliteHeart](https://github.com/huggingface/diffusers/pull/11297/) who helped us rewrite the [`AuraFlowTransformer2DModel`] class so that the above works for different resolutions ([PR](https://github.com/huggingface/diffusers/pull/11297/)).

0 commit comments

Comments
 (0)