From 55133b0910c2065a5b04e5495ab6d96226a17c9a Mon Sep 17 00:00:00 2001 From: Tair Date: Mon, 20 Jan 2025 16:19:30 +0000 Subject: [PATCH] return model id from TG info node --- nodes/tailored_model_info_node.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/nodes/tailored_model_info_node.py b/nodes/tailored_model_info_node.py index aec2b24..a04eaf6 100644 --- a/nodes/tailored_model_info_node.py +++ b/nodes/tailored_model_info_node.py @@ -11,8 +11,8 @@ class TailoredModelInfoNode(): } } - RETURN_TYPES = ("STRING", "INT", "INT", ) - RETURN_NAMES = ("generation_prefix", "default_fast", "default_steps_num", ) + RETURN_TYPES = ("STRING", "STRING","INT", "INT", ) + RETURN_NAMES = ("generation_prefix", "model_id", "default_fast", "default_steps_num", ) CATEGORY = "API Nodes" FUNCTION = "execute" # This is the method that will be executed @@ -30,6 +30,6 @@ class TailoredModelInfoNode(): training_version = response.json()["training_version"] default_fast = 1 if training_version == "light" else 0 default_steps_num = 8 if training_version == "light" else 30 - return (generation_prefix, default_fast, default_steps_num,) + return (generation_prefix, model_id, default_fast, default_steps_num,) else: raise Exception(f"Error: API request failed with status code {response.status_code} and text {response.text}")