Fix deprecated diffusers import

ionite34 · ionite34 · commit 77fe48735db6 · 2024-04-05T01:09:38.000-04:00
diff --git a/src/inference_core_nodes/layer_diffuse/lib_layerdiffusion/models.py b/src/inference_core_nodes/layer_diffuse/lib_layerdiffusion/models.py
@@ -7,7 +7,7 @@
 from typing import Optional, Tuple
 from diffusers.configuration_utils import ConfigMixin, register_to_config
 from diffusers.models.modeling_utils import ModelMixin
-from diffusers.models.unet_2d_blocks import UNetMidBlock2D, get_down_block, get_up_block
+from diffusers.models.unets.unet_2d_blocks import UNetMidBlock2D, get_down_block, get_up_block
 
 
 def check_diffusers_version():
