diffusers/tests/models/test_attention_dispatch.py at cli-tp

mirror of https://github.com/huggingface/diffusers.git synced 2026-10-27 21:33:02 +03:00

Files

Shivam Shrirao 80c7ed262a Fix flash/sage varlen prep under torch.compile with dynamic shapes (#14568 )

* Fix flash/sage varlen prep under torch.compile with dynamic shapes

* Handle zero-length sequences in varlen offsets

* Add varlen dynamic compile test

* Remove private helper test

2026-09-20 11:21:20 +05:30

9.0 KiB

Raw Permalink Blame History

View Raw

9.0 KiB Raw Permalink Blame History

9.0 KiB

Raw Permalink Blame History