Add new WebUI conversion state_dict_utils to __init__ utils (#6404)

apolinario · multimodalart · web-flow · commit bf725e044e0b · 2023-12-30T09:31:39.000-06:00
* Add new state_dict_utils to __init__ utils

* style

---------

Co-authored-by: multimodalart &lt;joaopaulo.passos+multimodal@gmail.com&gt;
diff --git a/src/diffusers/utils/__init__.py b/src/diffusers/utils/__init__.py
@@ -98,7 +98,9 @@
 )
 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,
 )

Original file line number	Diff line number	Diff line change
`@@ -98,7 +98,9 @@`
`98`	`98`	`)`
`99`	`99`	`from .pil_utils import PIL_INTERPOLATION, make_image_grid, numpy_to_pil, pt_to_pil`
`100`	`100`	`from .state_dict_utils import (`
	`101`	`+ convert_all_state_dict_to_peft,`
`101`	`102`	`convert_state_dict_to_diffusers,`
	`103`	`+ convert_state_dict_to_kohya,`
`102`	`104`	`convert_state_dict_to_peft,`
`103`	`105`	`convert_unet_state_dict_to_peft,`
`104`	`106`	`)`