Sayak Paul e4b8e7928b [Core] better support offloading when side loading is enabled. (#4855) ...

* better support offloading when side loading is enabled.

* load_textual_inversion

* better messaging for textual inversion.

* fixes

* address PR feedback.

* sdxl support.

* improve messaging

* recursive removal when cpu sequential offloading is enabled.

* add: lora tests

* recruse.

* add: offload tests for textual inversion.

2023-09-05 06:55:13 +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

relax tol attention conversion test (#3842)

2023-06-21 12:35:38 -07:00

test_layers_utils.py

Load Kohya-ss style LoRAs with auxilary states (#4147)

2023-07-26 00:24:19 +02:00

test_lora_layers.py

[Core] better support offloading when side loading is enabled. (#4855)

2023-09-05 06:55:13 +05:30

test_modeling_common_flax.py

[Tests] parallelize (#3078)

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

test_modeling_common.py

Clean up flaky behaviour on Slow CUDA Pytorch Push Tests (#4759)

2023-08-24 18:58:02 +05:30

test_models_prior.py

Complete set_attn_processor for prior and vae (#3796)

2023-06-15 17:42:49 +02:00

test_models_unet_1d.py

feat: add act_fn param to OutValueFunctionBlock (#3994)

2023-07-19 14:44:44 +02:00

test_models_unet_2d_condition.py

Test Cleanup Precision issues (#4812)

2023-09-01 17:58:37 +05:30

test_models_unet_2d_flax.py

Replace flake8 with ruff and update black (#2279)

2023-02-07 23:46:23 +01:00

test_models_unet_2d.py

Test Cleanup Precision issues (#4812)

2023-09-01 17:58:37 +05:30

test_models_unet_3d_condition.py

[Safetensors] Make safetensors the default way of saving weights (#4235)

2023-08-17 10:54:28 +05:30

test_models_vae_flax.py

[Tests] parallelize (#3078)

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

test_models_vae.py

Fix AutoencoderTiny encoder scaling convention (#4682)

2023-08-23 08:38:37 +05:30

test_models_vq.py

Complete set_attn_processor for prior and vae (#3796)

2023-06-15 17:42:49 +02:00

test_unet_2d_blocks.py

[Tests] parallelize (#3078)

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

test_unet_blocks_common.py

[Tests] parallelize (#3078)

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