From bf725e044e0b2fec29c074510c2d12d5ef66d2d1 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?apolin=C3=A1rio?= Date: Sat, 30 Dec 2023 09:31:39 -0600 Subject: [PATCH] Add new WebUI conversion state_dict_utils to __init__ utils (#6404) * Add new state_dict_utils to __init__ utils * style --------- Co-authored-by: multimodalart --- 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 181905bf5a..adc1cb8a3d 100644 --- a/src/diffusers/utils/__init__.py +++ b/src/diffusers/utils/__init__.py @@ -98,7 +98,9 @@ from .peft_utils import ( ) 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, )