Dan Saunders 5f8f817200 SP context manager update (#2699) ...

* utilize accelerate prepare_data_loader with patching

* lint

* cleanup, fix

* update to support DPO quirk

* coderabbit commits, cleanup, remove dead code

* fix

* move ring attn patching to sp ctx manager

* lint

* lint

* test fix

* test fix

2025-05-22 11:18:32 -04:00

16 KiB

Raw Permalink Blame History

View Raw