Restructure

__init__.py

@@ -1,4 +1,9 @@
-from .loaders import CoreMLLoaderUNet
+import os
+import sys
+
+sys.path.append(os.path.dirname(__file__))
+
+from coreml_suite import CoreMLLoaderUNet
 
 NODE_CLASS_MAPPINGS = {
     "CoreMLUNetLoader": CoreMLLoaderUNet,

coreml_suite/__init__.py

@@ -0,0 +1,3 @@
+from .loaders import CoreMLLoaderUNet
+
+__all__ = ["CoreMLLoaderUNet"]

coreml_suite/loaders.py

@@ -5,9 +5,9 @@ import folder_paths
 from comfy import supported_models_base, model_management
 from comfy.latent_formats import SD15
 from comfy.model_patcher import ModelPatcher
-from .logger import logger
-from .model import CoreMLModelWrapper
 
+from coreml_suite.logger import logger
+from coreml_suite.model import CoreMLModelWrapper
 
 class CoreMLLoader:
     PACKAGE_DIRNAME = ""

coreml_suite/model.py

@@ -5,7 +5,7 @@ from python_coreml_stable_diffusion.coreml_model import CoreMLModel
 
 from comfy.model_base import BaseModel
 
-from .utils import expand_inputs, extract_residual_kwargs
+from coreml_suite.utils import expand_inputs, extract_residual_kwargs
 
 
 class CoreMLModelWrapper(BaseModel):

coreml_suite/utils.py

@@ -3,7 +3,7 @@ from itertools import chain
 import numpy as np
 import torch
 
-from .logger import logger
+from coreml_suite.logger import logger
 
 
 def expand_inputs(inputs):