This commit is contained in:
Haoming 2026-03-16 12:18:20 +08:00
parent c9433ba95e
commit ad56e2fc7b
3 changed files with 1 additions and 10 deletions

View File

@ -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")

View File

@ -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")

View File

@ -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