Merge pull request #22 from userElaina/main

fix: `NoneType` object has no attribute `copy` in old version

.github/workflows/publish.yml

@@ -7,17 +7,19 @@ on:
     paths:
       - "pyproject.toml"
 
+permissions:
+  issues: write
+
 jobs:
   publish-node:
     name: Publish Custom Node to registry
     runs-on: ubuntu-latest
-    # if this is a forked repository. Skipping the workflow.
-    if: github.event.repository.fork == false
+    if: ${{ github.repository_owner == 'MinusZoneAI' }}
     steps:
       - name: Check out code
         uses: actions/checkout@v4
       - name: Publish Custom Node
-        uses: Comfy-Org/publish-node-action@main
+        uses: Comfy-Org/publish-node-action@v1
         with:
           ## Add your own personal access token to your Github Repository secrets and reference it here.
           personal_access_token: ${{ secrets.REGISTRY_ACCESS_TOKEN }}

__init__.py

@@ -293,6 +293,11 @@ class MZ_KohyaSSTrain_oldversion:
 
     def start(self, **kwargs):
         importlib.reload(mz_train_tools_core)
+        train_config = kwargs.get("train_config", "empty").copy()
+        if isinstance(train_config, dict):
+            for i in train_config:
+                if i not in kwargs:
+                    kwargs[i] = train_config[i]
         return mz_train_tools_core.MZ_KohyaSSTrain_call(kwargs)
 
 

hook_kohya_ss_run.py

@@ -34,8 +34,13 @@ import numpy as np
 import tempfile
 import safetensors.torch
 
-import hook_kohya_ss_utils
 
+import sys
+sys.path.append(os.path.dirname(__file__))
+try:
+    import hook_kohya_ss_utils
+except:
+    from . import hook_kohya_ss_utils
 
 other_config = {}
 original_save_model = None