Update onethingai.py
This commit is contained in:
+2
-2
@@ -26,7 +26,7 @@ class OneThingAILoader:
|
||||
|
||||
RETURN_TYPES = ("IMAGE", )
|
||||
FUNCTION = "generate"
|
||||
CATEGORY = "image generation"
|
||||
CATEGORY = "OneThingAI/image generation"
|
||||
|
||||
def generate(self, api_key, prompt, model="gpt4o", num_images=1, width=1024, height=1024, retries=3, timeout=8):
|
||||
# API endpoint
|
||||
@@ -119,4 +119,4 @@ NODE_CLASS_MAPPINGS = {
|
||||
# Add descriptions
|
||||
NODE_DISPLAY_NAME_MAPPINGS = {
|
||||
"OneThingAILoader": "OneThingAI Image Generator"
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user