Merge pull request #85 from yaosheng216/patch-17

Update pyproject.toml

.github/workflows/publish.yml

@@ -0,0 +1,24 @@
+name: Publish to Comfy registry
+on:
+  workflow_dispatch:
+  push:
+    branches:
+      - main
+      - master
+    paths:
+      - "scepter/workflow/pyproject.toml"
+
+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 
+    steps:
+      - name: Check out code
+        uses: actions/checkout@v4
+      - name: Publish Custom Node
+        uses: Comfy-Org/publish-node-action@main
+        with:
+          ## Add your own personal access token to your Github Repository secrets and reference it here.
+          personal_access_token: ${{ secrets.REGISTRY_ACCESS_TOKEN }}

requirements/framework.txt

@@ -1,6 +1,5 @@
 albumentations
 beautifulsoup4
-bezier
 einops
 modelscope[framework]
 ms-swift

scepter/studio/self_train/self_train.py

@@ -3,9 +3,10 @@
 import os
 
 import gradio as gr
-import scepter
+
 from scepter.modules.utils.config import Config
 from scepter.modules.utils.file_system import FS
+from scepter.modules.utils.import_utils import get_dirname
 from scepter.studio.self_train.self_train_ui.model_ui import ModelUI
 from scepter.studio.self_train.self_train_ui.trainer_ui import TrainerUI
 from scepter.studio.self_train.utils.config_parser import get_all_config
@@ -51,7 +52,7 @@ class SelfTrainUI():
 
 if __name__ == '__main__':
     st_ins = SelfTrainUI(os.path.join(
-        scepter.dirname, 'scepter/methods/studio/self_train/self_train.yaml'),
+        get_dirname(), 'scepter/methods/studio/self_train/self_train.yaml'),
                          is_debug=True,
                          language='zh',
                          root_work_dir='./cache')

scepter/tools/webui.py

@@ -9,7 +9,6 @@ import sys
 
 import gradio as gr
 
-import scepter
 from scepter.modules.utils.config import Config
 from scepter.modules.utils.file_system import FS
 from scepter.modules.utils.logger import get_logger, init_logger
@@ -95,7 +94,7 @@ if __name__ == '__main__':
                         '') if args.language == 'en' else info['NAME']
         ifid = info['IFID']
         if not FS.exists(info['CONFIG']):
-            info['CONFIG'] = os.path.join(os.path.dirname(scepter.dirname),
+            info['CONFIG'] = os.path.join(os.path.dirname(get_dirname()),
                                           info['CONFIG'])
         if not FS.exists(info['CONFIG']):
             raise f"{info['CONFIG']} doesn't exist."

scepter/workflow/pyproject.toml

@@ -0,0 +1,15 @@
+[project]
+name = "scepter"
+description = "Custom nodes for various visual generation and editing tasks using Scepter."
+version = "1.4.0"
+license = {file = "LICENSE"}
+dependencies = ["-r requirements/framework.txt", "-r requirements/scepter_studio.txt"]
+
+[project.urls]
+Repository = "https://github.com/modelscope/scepter"
+#  Used by Comfy Registry https://comfyregistry.org
+
+[tool.comfy]
+PublisherId = "scepter"
+DisplayName = "scepter"
+Icon = "https://github.com/modelscope/scepter/blob/main/asset/images/icon.png"