Sayak Paul 85a916bb8b make group offloading work with disk/nvme transfers (#11682) ...

* start implementing disk offloading in group.

* delete diff file.

* updates.patch

* offload_to_disk_path

* check if safetensors already exist.

* add test and clarify.

* updates

* update todos.

* update more docs.

* update docs

2025-06-19 18:09:30 +05:30

..

autoencoders

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

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

transformers

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

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

make group offloading work with disk/nvme transfers (#11682)

2025-06-19 18:09:30 +05:30

test_models_auto.py

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

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