diffusers

mirror of https://github.com/huggingface/diffusers.git synced 2026-01-27 17:22:53 +03:00

Files

Sayak Paul 365e8461ac [SDXL DreamBooth LoRA] add support for text encoder fine-tuning (#4097 )

* Allow low precision sd xl

* finish

* finish

* feat: initial draft for supporting text encoder lora finetuning for SDXL DreamBooth

* fix: variable assignments.

* add: autocast block.

* add debugging

* vae dtype hell

* fix: vae dtype hell.

* fix: vae dtype hell 3.

* clean up

* lora text encoder loader.

* fix: unwrapping models.

* add: tests.

* docs.

* handle unexpected keys.

* fix vae dtype in the final inference.

* fix scope problem.

* fix: save_model_card args.

* initialize: prefix to None.

* fix: dtype issues.

* apply gixes.

* debgging.

* debugging

* debugging

* debugging

* debugging

* debugging

* add: fast tests.

* pre-tokenize.

* address: will's comments.

* fix: loader and tests.

* fix: dataloader.

* simplify dataloader.

* length.

* simplification.

* make style && make quality

* simplify state_dict munging

* fix: tests.

* fix: state_dict packing.

* Apply suggestions from code review

Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com>

---------

Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com>

2023-07-25 05:35:48 +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

Refactor full determinism (#3485 )