diff --git a/debug_imports.py b/debug_imports.py new file mode 100644 index 0000000..0519ecb --- /dev/null +++ b/debug_imports.py @@ -0,0 +1 @@ + \ No newline at end of file diff --git a/node_category_report.txt b/node_category_report.txt new file mode 100644 index 0000000..53a8f13 --- /dev/null +++ b/node_category_report.txt @@ -0,0 +1 @@ +=== UniversalToolkit 节点分组属性清单 === diff --git a/nodes/image/empty_unit_generator.py b/nodes/image/empty_unit_generator.py index d571a86..379bc5b 100644 --- a/nodes/image/empty_unit_generator.py +++ b/nodes/image/empty_unit_generator.py @@ -106,8 +106,8 @@ class EmptyUnitGenerator_UTK: "optional": {}, } - RETURN_TYPES = ("IMAGE", "MASK", "LATENT", "INT", "INT") - RETURN_NAMES = ("image", "mask", "latent", "width", "height") + RETURN_TYPES = ("IMAGE", "MASK", "LATENT", "INT", "INT", "INT") + RETURN_NAMES = ("image", "mask", "latent", "width", "height", "batch_num") FUNCTION = "generate" def generate( @@ -159,7 +159,7 @@ class EmptyUnitGenerator_UTK: ), "batch_index_list": None, } - return images, masks, latent, w, h + return images, masks, latent, w, h, batch # Node mappings diff --git a/nodes/tools/lora_info_db.json b/nodes/tools/lora_info_db.json index 953719e..23bd567 100644 --- a/nodes/tools/lora_info_db.json +++ b/nodes/tools/lora_info_db.json @@ -59,5 +59,12 @@ "examplePrompt": null, "baseModel": "Flux.1 Kontext", "metaInfo": "{\n \"format\": \"pt\"\n}" + }, + "Kontext\\HDR360.safetensors": { + "output": "URL: https://civitai.com/models/682349\nTriggers: create a 360 hdr\nBase Model: Flux.1 Kontext\n\nExamples:\n\nOutput: https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/7d251dc9-19bd-4bab-9d5b-e1611022e547/width=1800/85418192.mp4\n\n\nOutput: https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/ef8c26e5-aec3-4d75-8a15-3831ee00ea6d/width=1800/85405285.jpeg\n\n\nOutput: https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/ab9bb167-62c0-4443-aedc-26be9fd4bbe2/width=736/85405295.jpeg\n\n\nOutput: https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/66bfe2a9-91da-4f5e-8e89-d6690cb701d0/width=1800/85405332.jpeg\n\n", + "trainedWords": "create a 360 hdr", + "examplePrompt": null, + "baseModel": "Flux.1 Kontext", + "metaInfo": "{\n \"name\": \"HDR360\",\n \"sshs_legacy_hash\": \"6bb22ed6\",\n \"format\": \"pt\",\n \"training_info\": \"{\\\"step\\\": 5500, \\\"epoch\\\": 56}\",\n \"sshs_model_hash\": \"d904a130eb54877361ec4cd9ebd82c3f937c1128a56ea68115c5db0ba1291638\",\n \"version\": \"1.0\",\n \"ss_base_model_version\": \"sd_1.5\",\n \"ss_output_name\": \"HDR360\",\n \"software\": \"{\\\"name\\\": \\\"ai-toolkit\\\", \\\"repo\\\": \\\"https://github.com/ostris/ai-toolkit\\\", \\\"version\\\": \\\"0.3.3\\\"}\"\n}" } } \ No newline at end of file diff --git a/pyproject.toml b/pyproject.toml index 9743eb7..b40e88b 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,7 +1,7 @@ [project] name = "universaltoolkit" description = "A comprehensive toolkit based on ComfyUI, providing image, mask, audio, and tools nodes, fully modular and v3 compatible." -version = "1.2.8" +version = "1.2.9" license = {file = "LICENSE"} dependencies = [ "torch",