Small fix to support diffusers 0.12.0

Diffusers 0.12.0 was released yesterday.
1 year ago · 5c16c73796
parent fb1fa01c9d
commit 5c16c73796
1 changed files with 1 additions and 1 deletions
--- a/python_coreml_stable_diffusion/unet.py
+++ b/python_coreml_stable_diffusion/unet.py
@ -6,7 +6,7 @@
 from python_coreml_stable_diffusion.layer_norm import LayerNormANE

 from diffusers.configuration_utils import ConfigMixin, register_to_config
-from diffusers.modeling_utils import ModelMixin
+from diffusers import ModelMixin

 from enum import Enum