update nodes

.gitignore

@@ -0,0 +1 @@
+*cache*

Custom_ComfyUI-YoloWorld-EfficientSAM/__init__.py

@@ -1,5 +0,0 @@
-from . import YOLO_WORLD_EfficientSAM
-from . import YOLO_WORLD_SEGS
-
-NODE_CLASS_MAPPINGS = {**YOLO_WORLD_EfficientSAM.NODE_CLASS_MAPPINGS, **YOLO_WORLD_SEGS.NODE_CLASS_MAPPINGS}
-NODE_DISPLAY_NAME_MAPPINGS = {**YOLO_WORLD_EfficientSAM.NODE_DISPLAY_NAME_MAPPINGS, **YOLO_WORLD_SEGS.NODE_DISPLAY_NAME_MAPPINGS}

Custom_ComfyUI_YoloWorld_EfficientSAM/YOLO_WORLD_EfficientSAM.py

@@ -129,7 +129,7 @@ class Yoloworld_ESAM_Zho:
     RETURN_TYPES = ("IMAGE", "MASK", "STRING", "STRING")
     RETURN_NAMES = ("IMAGE", "MASK", "BBOX", "CATEGORIES")
     FUNCTION = "yoloworld_esam_image"
-    CATEGORY = "🔎YOLOWORLD_ESAM"
+    CATEGORY = "🔎YOLOWORLD_ESAM_DUCI"
                        
     def yoloworld_esam_image(self, image, yolo_world_model, esam_model, categories, confidence_threshold, iou_threshold, box_thickness, text_thickness, text_scale, with_segmentation, mask_combined, with_confidence, with_class_agnostic_nms, mask_extracted, mask_extracted_index):
         categories = process_categories(categories)

Custom_comfyui-llm-assistant/__init__.py

@@ -1,3 +0,0 @@
-from .AssistantNodes import NODE_CLASS_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS
-
-__all__ = ['NODE_CLASS_MAPPINGS', 'NODE_DISPLAY_NAME_MAPPINGS']

Custom_comfyui-llm-assistant/requirements.txt

@@ -1,2 +0,0 @@
-pyyaml
-openai

Custom_comfyui_llm_assistant/AssistantNodes.py

@@ -2,9 +2,6 @@ from .utils import *
 
 
 class GenerateStableDiffsutionPromptLLM:
-    def __init__(self):
-        pass
-
     @classmethod
     def INPUT_TYPES(cls):
         data = load_config()
@@ -101,7 +98,7 @@ class GenerateStableDiffsutionPromptLLM:
 
     FUNCTION = "generateStableDiffsutionPrompt"
 
-    CATEGORY = "LLM"
+    CATEGORY = "cropcropcropcrop"
 
     def generateStableDiffsutionPrompt(self, object1, desc_obj1, object2, desc_obj2 ,template_system, template_user, stop, response_pattern,  temperature, max_tokens, model_name):
         config = load_config()
@@ -331,13 +328,13 @@ class ChatWithLLM:
 
 
 NODE_CLASS_MAPPINGS = {
-    "Generate Stable Diffsution Prompt With LLM": GenerateStableDiffsutionPromptLLM,
-    "Translate Text With LLM": TranslateTextLLM,
-    "Chat With LLM": ChatWithLLM,
+    "Generate Stable Diffsution Prompt With LLM duci": GenerateStableDiffsutionPromptLLM,
+    # "Translate Text With LLM": TranslateTextLLM,
+    # "Chat With LLM": ChatWithLLM,
 }
 
 NODE_DISPLAY_NAME_MAPPINGS = {
-    "Generate Stable Diffsution Prompt With LLM": "Custom Generate Stable Diffsution Prompt With LLM",
-    "Translate Text With LLM": "Translate Text With LLM",
-    "Chat With LLM": "Chat With LLM",
+    "Generate Stable Diffsution Prompt With LLM duci": "HUHU",
+    # "Translate Text With LLM": "Translate Text With LLM",
+    # "Chat With LLM": "Chat With LLM",
 }

Custom_comfyui_llm_assistant/requirements.txt

@@ -0,0 +1,2 @@
+# pyyaml
+# openai

README.md

@@ -32,7 +32,7 @@ Here, I introduce a complete pipeline that takes two images as input, allowing y
 
 2. Install the required packages:
     ```bash
-    pip install -r requirements.txt
+    pip install -r myreqs.txt
     ```
 
 3. Clone these amazing repositories and follow their instructions:

__init__.py

@@ -0,0 +1,14 @@
+# from .Custom_ComfyUI_YoloWorld_EfficientSAM import YOLO_WORLD_EfficientSAM
+# from .Custom_ComfyUI_YoloWorld_EfficientSAM import YOLO_WORLD_SEGS
+from .Custom_comfyui_llm_assistant import AssistantNodes
+from .CROP_OBJECT import crop
+
+# NODE_CLASS_MAPPINGS = {**YOLO_WORLD_EfficientSAM.NODE_CLASS_MAPPINGS, **YOLO_WORLD_SEGS.NODE_CLASS_MAPPINGS, **AssistantNodes.NODE_CLASS_MAPPINGS, **crop.NODE_CLASS_MAPPINGS}
+# NODE_DISPLAY_NAME_MAPPINGS = {**YOLO_WORLD_EfficientSAM.NODE_DISPLAY_NAME_MAPPINGS, **YOLO_WORLD_SEGS.NODE_DISPLAY_NAME_MAPPINGS, **AssistantNodes.NODE_DISPLAY_NAME_MAPPINGS, **crop.NODE_DISPLAY_NAME_MAPPINGS}
+
+
+NODE_CLASS_MAPPINGS = {**AssistantNodes.NODE_CLASS_MAPPINGS, **crop.NODE_CLASS_MAPPINGS}
+NODE_DISPLAY_NAME_MAPPINGS = {**AssistantNodes.NODE_DISPLAY_NAME_MAPPINGS, **crop.NODE_DISPLAY_NAME_MAPPINGS}
+
+print(NODE_CLASS_MAPPINGS)
+print(NODE_DISPLAY_NAME_MAPPINGS)

myreqs.txt

@@ -1,41 +0,0 @@
-aiohttp==3.10.4
-aria2p==0.12.0
-einops==0.8.0
-GitPython==3.1.43
-intel_extension_for_pytorch==2.4.0
-kornia==0.7.3
-matrix_client==0.4.0
-nltk==3.9.1
-numpy==2.1.0
-ollama==0.3.1
-openai==1.42.0
-opencv_python==4.8.0.76
-opencv_python_headless==4.10.0.84
-Pillow==10.4.0
-psutil==5.9.0
-pygit2==1.15.1
-PyGithub==2.3.0
-pytest==8.3.2
-PyYAML==6.0.2
-Requests==2.32.3
-rich==13.7.1
-safetensors==0.4.4
-scipy==1.14.1
-sentencepiece==0.2.0
-scikit-image=0.24.0
-spandrel==0.3.4
-spandrel_extra_arches==0.1.1
-supervision==0.22.0
-torch==2.4.0
-torch_directml==0.2.4.dev240815
-torchaudio==2.4.0
-torchsde==0.2.6
-torchvision==0.19.0
-tqdm==4.66.4
-transformers==4.44.0
-typer==0.12.4
-typing_extensions==4.12.2
-ultralytics==8.2.78
-websocket_client==1.8.0
-xformers==0.0.27.post2
-

requirements.txt

@@ -1,41 +1,27 @@
-aiohttp
-aria2p
-einops
-GitPython
-intel_extension_for_pytorch
-kornia
-matrix_client
-nltk
-numpy
-ollama
-openai
-opencv_python
-opencv_python_headless
-Pillow
-psutil
-pygit2
-PyGithub
-pytest
-PyYAML
-Requests
-rich
-safetensors
-scipy
-sentencepiece
-scikit-image
-spandrel
-spandrel_extra_arches
-supervision
-torch
-torch_directml
-torchaudio
-torchsde
-torchvision
-tqdm
-transformers
-typer
-typing_extensions
-ultralytics
-websocket_client
-xformers
-
+einops==0.8.0
+spandrel==0.3.4
+kornia==0.7.3
+nltk==3.9.1
+ollama==0.3.1
+openai==1.42.0
+opencv_python==4.8.0.76
+opencv_python_headless==4.10.0.84
+Pillow==10.4.0
+psutil==5.9.0
+PyYAML==6.0.2
+Requests==2.31.0
+safetensors==0.4.4
+scipy==1.14.1
+sentencepiece==0.2.0
+scikit-image==0.24.0
+supervision==0.22.0
+torch==2.4.0
+torchaudio==2.4.0
+torchsde==0.2.6
+torchvision==0.19.0
+tqdm==4.66.4
+transformers==4.44.0
+ultralytics==8.2.78
+websocket_client==1.8.0
+xformers==0.0.27.post2
+inference-gpu[yolo-world]==0.16.2