diff --git a/backend/args.py b/backend/args.py index 25d0599d..94cfaa02 100644 --- a/backend/args.py +++ b/backend/args.py @@ -69,7 +69,6 @@ parser.add_argument("--flash", action="store_true", help="install flash_attn") parser.add_argument("--xformers", action="store_true", help="install xformers") parser.add_argument("--nunchaku", action="store_true", help="install nunchaku for SVDQ inference") parser.add_argument("--bnb", action="store_true", help="install bitsandbytes for 4-bit inference") -parser.add_argument("--ck", action="store_true", help="install comfy-kitchen for Mixed Precision inference") parser.add_argument("--onnxruntime-gpu", action="store_true", help="install nightly onnxruntime-gpu with cu130 support") parser.add_argument("--disable-sage", action="store_true", help="disable sageattention") diff --git a/modules/launch_utils.py b/modules/launch_utils.py index 011294d6..7ec2bae5 100644 --- a/modules/launch_utils.py +++ b/modules/launch_utils.py @@ -292,7 +292,6 @@ def prepare_environment(): torch_command = os.environ.get("TORCH_COMMAND", f"pip install torch==2.10.0+cu130 torchvision==0.25.0+cu130 --extra-index-url {torch_index_url}") xformers_package = os.environ.get("XFORMERS_PACKAGE", f"xformers==0.0.34 --extra-index-url {torch_index_url}") bnb_package = os.environ.get("BNB_PACKAGE", "bitsandbytes==0.49.1") - ck_package = os.environ.get("CK_PACKAGE", "comfy-kitchen==0.2.8") packaging_package = os.environ.get("PACKAGING_PACKAGE", "packaging==26.0") gradio_package = os.environ.get("GRADIO_PACKAGE", "gradio==4.40.0 gradio_rangeslider==0.0.8") @@ -411,14 +410,6 @@ def prepare_environment(): else: startup_timer.record("install bitsandbytes") - if args.ck and not is_installed("comfy_kitchen"): - try: - run_pip(f"install {ck_package}", "comfy-kitchen") - except RuntimeError: - print("Failed to install comfy-kitchen; Please manually install it") - else: - startup_timer.record("install comfy-kitchen") - if not is_installed("ngrok") and args.ngrok: run_pip("install ngrok", "ngrok") startup_timer.record("install ngrok") diff --git a/requirements.txt b/requirements.txt index c37c6e37..909cce62 100644 --- a/requirements.txt +++ b/requirements.txt @@ -2,6 +2,7 @@ GitPython==3.1.46 Pillow==12.1.0 accelerate==1.12.0 audioop-lts==0.2.2;python_version>="3.13" +comfy-kitchen==0.2.8 diffusers==0.36.0 diskcache==5.6.3 einops==0.8.2