Revert "fix: version bug resolve for ComfyUI blocked at 0.3.27 (#244)" (#259)

This reverts commit 064c4df.

Author: eliteprox

docker/Dockerfile.base

@@ -64,7 +64,7 @@ RUN conda run -n comfystream pip install --no-cache-dir \
     ${TensorRT_ROOT}/python/tensorrt-10.9.0.34-cp311-none-linux_x86_64.whl
 
 # Clone ComfyUI
-RUN git clone --branch v0.3.39 --depth 1 https://github.com/comfyanonymous/ComfyUI.git /workspace/ComfyUI
+RUN git clone --branch v0.3.27 --depth 1 https://github.com/comfyanonymous/ComfyUI.git /workspace/ComfyUI
 
 # Copy only files needed for setup
 COPY ./src/comfystream/scripts /workspace/comfystream/src/comfystream/scripts
@@ -105,7 +105,4 @@ RUN conda config --set auto_activate_base false && \
 # Set comfystream environment as default
 RUN echo "conda activate comfystream" >> ~/.bashrc
 
-# Set /workspace/ComfyUI to PYTHONPATH
-RUN echo "export PYTHONPATH=/workspace/ComfyUI" >> ~/.bashrc
-
 WORKDIR /workspace/comfystream

pyproject.toml

@@ -9,7 +9,7 @@ version = "0.1.2"
 license = { file = "LICENSE" }
 dependencies = [
     "asyncio",
-    "comfyui @ git+https://github.com/livepeer/ComfyUI.git@6a39b5b016cc108cbc5dda2fd4b32a51161ab150",
+    "comfyui @ git+https://github.com/hiddenswitch/ComfyUI.git@ce3583ad42c024b8f060d0002cbe20c265da6dc8",
     "aiortc",
     "aiohttp",
     "aiohttp_cors",

requirements.txt

@@ -1,5 +1,5 @@
 asyncio
-comfyui @ git+https://github.com/livepeer/ComfyUI.git@6a39b5b016cc108cbc5dda2fd4b32a51161ab150
+comfyui @ git+https://github.com/hiddenswitch/ComfyUI.git@ce3583ad42c024b8f060d0002cbe20c265da6dc8
 aiortc
 aiohttp
 aiohttp_cors

src/comfystream/client.py

@@ -5,9 +5,9 @@
 from comfystream import tensor_cache
 from comfystream.utils import convert_prompt
 
-from hiddenswitch_comfy.api.components.schema.prompt import PromptDictInput
-from hiddenswitch_comfy.cli_args_types import Configuration
-from hiddenswitch_comfy.client.embedded_comfy_client import EmbeddedComfyClient
+from comfy.api.components.schema.prompt import PromptDictInput
+from comfy.cli_args_types import Configuration
+from comfy.client.embedded_comfy_client import EmbeddedComfyClient
 
 logger = logging.getLogger(__name__)
 
@@ -114,7 +114,7 @@ async def get_available_nodes(self):
             return {}
 
         try:
-            from hiddenswitch_comfy.nodes.package import import_all_nodes_in_workspace
+            from comfy.nodes.package import import_all_nodes_in_workspace
             nodes = import_all_nodes_in_workspace()
 
             all_prompts_nodes_info = {}

src/comfystream/utils.py

@@ -1,7 +1,7 @@
 import copy
 
 from typing import Dict, Any
-from hiddenswitch_comfy.api.components.schema.prompt import Prompt, PromptDictInput
+from comfy.api.components.schema.prompt import Prompt, PromptDictInput
 
 
 def create_load_tensor_node():

test/test_utils.py

@@ -1,6 +1,6 @@
 import pytest
 
-from hiddenswitch_comfy.api.components.schema.prompt import Prompt
+from comfy.api.components.schema.prompt import Prompt
 from comfystream.utils import convert_prompt