fix(builder): file custom nodes types

2024-01-11 12:23:32 +08:00 · 2024-01-11 12:23:32 +08:00 · a4b8dbf1d3
commit a4b8dbf1d3
parent 745d3f9d35
1 changed files with 5 additions and 1 deletions
--- a/builder/modal-builder/src/main.py
+++ b/builder/modal-builder/src/main.py
@ -140,11 +140,15 @@ class GitCustomNodes(BaseModel):
    hash: str
    disabled: bool

+class FileCustomNodes(BaseModel):
+    filename: str
+    disabled: bool
+

 class Snapshot(BaseModel):
    comfyui: str
    git_custom_nodes: Dict[str, GitCustomNodes]
-
+    file_custom_nodes: List[FileCustomNodes]

 class Model(BaseModel):
    name: str