[hot fix]apply llm-vision tempture

xlinx · Aug 28, 2024 · 36d8958 · 36d8958
1 parent 374c0b7
commit 36d8958
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/auto_prompt_llm.py b/auto_prompt_llm.py
@@ -218,12 +218,12 @@ def call_llm_eye_open(clip,
                                                {'type': 'image_url', 'image_url': {'url': f'{base64_image}'}}
                                            ]}
                                        ],
-                                       'max_tokens': f'{llm_text_max_token}',
-                                       'temperature': f'{llm_text_tempture}',
+                                       'max_tokens': f'{llm_vision_max_token}',
+                                       'temperature': f'{llm_vision_tempture}',
                                        'stream': f'{False}',
                                    }
                                    ).json()
-        # print(f'[][][]{completion}')
+        print(f'[][][]{completion}')
         # pprint.pprint(completion)
         result = completion['choices'][0]['message']['content']
     except Exception as e:
@@ -474,7 +474,7 @@ def INPUT_TYPES(cls):
                 "llm_api_model_name": ("STRING", {"multiline": False, "default": "llama3.1"}),
 
                 "llm_vision_max_token": ("INT", {"default": 50, "min": 10, "max": 1024, "step": 1}),
-                "llm_vision_tempture": ("FLOAT", {"default": 0.8, "min": -2.0, "max": 2.0, "step": 0.01}),
+                "llm_vision_tempture": ("FLOAT", {"default": 0.2, "min": -2.0, "max": 2.0, "step": 0.01}),
                 "llm_vision_system_prompt": (
                     "STRING", {"multiline": True, "dynamicPrompts": True, "default": dafault_llm_sys_prompt_vision}),
                 "llm_vision_ur_prompt": (
@@ -575,7 +575,7 @@ def INPUT_TYPES(cls):
                     "STRING", {"multiline": True, "dynamicPrompts": True, "default": dafault_llm_user_prompt}),
 
                 "llm_vision_max_token": ("INT", {"default": 50, "min": 10, "max": 1024, "step": 1}),
-                "llm_vision_tempture": ("FLOAT", {"default": 0.8, "min": -2.0, "max": 2.0, "step": 0.01}),
+                "llm_vision_tempture": ("FLOAT", {"default": 0.2, "min": -2.0, "max": 2.0, "step": 0.01}),
                 "llm_vision_system_prompt": (
                     "STRING", {"multiline": True, "dynamicPrompts": True, "default": dafault_llm_sys_prompt_vision}),
                 "llm_vision_ur_prompt": (

diff --git a/pyproject.toml b/pyproject.toml
@@ -1,7 +1,7 @@
 [project]
 name = "comfyui-decadetw-auto-prompt-llm"
 description = "NODES: Auto-LLM-Text-Vision, Auto-LLM-Text, Auto-LLM-Vision"
-version = "1.20240827.2"
+version = "1.20240827.3"
 license = {text = "MIT License"}
 
 [project.urls]