From 9e1b2710e527fe5e8ed68027315922e3e3ef64af Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?apolin=C3=A1rio?= Date: Sat, 30 Dec 2023 09:03:35 -0600 Subject: [PATCH 1/2] Add new state_dict_utils to __init__ utils --- src/diffusers/utils/__init__.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/src/diffusers/utils/__init__.py b/src/diffusers/utils/__init__.py index 181905bf5ac0..89cadca575db 100644 --- a/src/diffusers/utils/__init__.py +++ b/src/diffusers/utils/__init__.py @@ -101,6 +101,8 @@ convert_state_dict_to_diffusers, convert_state_dict_to_peft, convert_unet_state_dict_to_peft, + convert_all_state_dict_to_peft, + convert_state_dict_to_kohya, ) From 518171600d3eb82fc4f4c84b81dd7564b02728dc Mon Sep 17 00:00:00 2001 From: multimodalart Date: Sat, 30 Dec 2023 12:04:44 -0300 Subject: [PATCH 2/2] style --- src/diffusers/utils/__init__.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/diffusers/utils/__init__.py b/src/diffusers/utils/__init__.py index 89cadca575db..adc1cb8a3dfb 100644 --- a/src/diffusers/utils/__init__.py +++ b/src/diffusers/utils/__init__.py @@ -98,11 +98,11 @@ ) from .pil_utils import PIL_INTERPOLATION, make_image_grid, numpy_to_pil, pt_to_pil from .state_dict_utils import ( + convert_all_state_dict_to_peft, convert_state_dict_to_diffusers, + convert_state_dict_to_kohya, convert_state_dict_to_peft, convert_unet_state_dict_to_peft, - convert_all_state_dict_to_peft, - convert_state_dict_to_kohya, )