diff --git a/workflows/preview_server/initialize_comfy_extensions.py b/workflows/preview_server/initialize_comfy_extensions.py index bae7c54..adad2b8 100644 --- a/workflows/preview_server/initialize_comfy_extensions.py +++ b/workflows/preview_server/initialize_comfy_extensions.py @@ -16,8 +16,9 @@ sys.path.append(comfy_path) from server import PromptServer from execution import PromptQueue -from main import init_custom_nodes, cuda_malloc_warning, prompt_worker, execute_prestartup_script +from main import cuda_malloc_warning, prompt_worker, execute_prestartup_script import folder_paths +from nodes import init_extra_nodes execute_prestartup_script() @@ -27,7 +28,7 @@ async def run(): server = PromptServer(loop) q = PromptQueue(server) - init_custom_nodes() + init_extra_nodes() cuda_malloc_warning() diff --git a/workflows/preview_server/preview_server.py b/workflows/preview_server/preview_server.py index cbde3e8..a1bf654 100644 --- a/workflows/preview_server/preview_server.py +++ b/workflows/preview_server/preview_server.py @@ -4,7 +4,6 @@ import time import shutil from pathlib import Path -import asyncio import uuid import os import json @@ -15,7 +14,6 @@ import mimetypes from fastapi import FastAPI, File, UploadFile, HTTPException, Form, Request, Response import requests from jsonpath_ng import jsonpath, parse -from moviepy.editor import ImageClip from fastapi.middleware.cors import CORSMiddleware import urllib import websockets @@ -399,7 +397,7 @@ if __name__ == "__main__": signal.signal(signal.SIGINT, signal_handler) signal.signal(signal.SIGTERM, signal_handler) - comfyui_process = subprocess.Popen(["venv/bin/python", "main.py", "--highvram", "--disable-metadata"], cwd=COMFY_ROOT_DIRECTORY) + comfyui_process = subprocess.Popen(["../venv/bin/python", "main.py", "--highvram", "--disable-metadata"], cwd=COMFY_ROOT_DIRECTORY) # block until comfyUI server is up diff --git a/workflows/preview_server/server_requirements.txt b/workflows/preview_server/server_requirements.txt index 0523c7c..16ff43f 100644 --- a/workflows/preview_server/server_requirements.txt +++ b/workflows/preview_server/server_requirements.txt @@ -4,5 +4,4 @@ fastapi~=0.110.0 python-multipart~=0.0.9 pydantic~=2.6.4 jsonpath-ng~=1.6.1 -moviepy websockets \ No newline at end of file