Rearrange stuff

2024-06-28 15:52:53 +02:00
parent b326b3d3b9
commit c3038501eb
8 changed files with 79 additions and 110 deletions
@@ -3,7 +3,7 @@ import sys

 sys.path.append(os.path.dirname(__file__))

-from coreml_suite import CoreMLLoaderUNet, CoreMLSampler
+from coreml_suite.nodes import CoreMLLoaderUNet, CoreMLSampler

 NODE_CLASS_MAPPINGS = {
    "CoreMLUNetLoader": CoreMLLoaderUNet,