Update python_coreml_stable_diffusion/unet.py

Co-authored-by: Pedro Cuenca <pedro@huggingface.co>
1 year ago · b3d0d01ec6
parent 51d3fd8c16
commit b3d0d01ec6
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.models.modeling_utils import ModelMixin
+from diffusers import ModelMixin

 from enum import Enum