nodes/Bria-AI-ComfyUI-BRIA-API
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Merge pull request #9 from Bria-AI/ranges

Browse Source 
return model id from TG info node
This commit is contained in:
BriaOr
2025-01-21 10:22:52 +02:00
committed by GitHub
parent da4c773cb7 55133b0910
commit 709d16cb72
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
nodes/tailored_model_info_node.py
+3 -3
View File
@@ -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}")