Sayak Paul 2d3d376bc0 Fix unique memory address when doing group-offloading with disk (#11767) ...

* fix memory address problem

* add more tests

* updates

* updates

* update

* _group_id = group_id

* update

* Apply suggestions from code review

Co-authored-by: Dhruv Nair <dhruv.nair@gmail.com>

* update

* update

* update

* fix

---------

Co-authored-by: Dhruv Nair <dhruv.nair@gmail.com>

2025-07-09 21:29:34 +05:30

..

autoencoders

[chore] change to 2025 licensing for remaining (#11741)

2025-06-18 20:56:00 +05:30

transformers

[tests] add test for hotswapping + compilation on resolution changes (#11825)

2025-07-01 09:40:34 +05:30

unets

[chore] change to 2025 licensing for remaining (#11741)

2025-06-18 20:56:00 +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

[Tests] Fix more tests sayak (#10359)

2024-12-23 19:11:21 +05:30

test_layers_utils.py

[chore] change to 2025 licensing for remaining (#11741)

2025-06-18 20:56:00 +05:30

test_modeling_common_flax.py

[Tests] parallelize (#3078)

2023-04-13 13:32:57 +01:00

test_modeling_common.py

Fix unique memory address when doing group-offloading with disk (#11767)

2025-07-09 21:29:34 +05:30

test_models_auto.py

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

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