Merge pull request #13 from LykosAI/fix-diffusers-import

Fix deprecated diffusers import

Author: ionite34

src/inference_core_nodes/__init__.py

@@ -1,6 +1,6 @@
 __all__ = ("__version__", "NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS")
 
-__version__ = "0.4.0"
+__version__ = "0.4.1"
 
 
 def _get_node_mappings():

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():