Kimbing Ng 3c05b9f71c Fixes #12673. record_stream in group offloading is not working properly (#12721) ...

* Fixes #12673.

    Wrong default_stream is used. leading to wrong execution order when record_steram is enabled.

* update

* Update test

---------

Co-authored-by: Sayak Paul <spsayakpaul@gmail.com>

2025-12-03 11:37:11 +05:30

..

autoencoders

skip autoencoderdl layerwise casting memory (#12647)

2025-11-13 12:56:22 +05:30

transformers

Hunyuanvideo15 (#12696)

2025-11-30 20:27:59 -10:00

unets

[Refactor] Move testing utils out of src (#12238)

2025-08-28 19:53:02 +05:30

__init__.py

[Tests] Improve unet / vae tests (#1018)

2022-10-28 13:43:26 +02:00

test_activations.py

move activation dispatches into helper function (#3656)

2023-06-05 12:30:48 -07:00

test_attention_processor.py

[Refactor] Move testing utils out of src (#12238)

2025-08-28 19:53:02 +05:30

test_layers_utils.py

[Refactor] Move testing utils out of src (#12238)

2025-08-28 19:53:02 +05:30

test_modeling_common.py

Fixes #12673. record_stream in group offloading is not working properly (#12721)

2025-12-03 11:37:11 +05:30

test_models_auto.py

[Feature] AutoModel can load components using model_index.json (#11401)

2025-05-26 14:06:36 +05:30