--- setup.py.orig 2025-03-01 00:00:00 UTC +++ setup.py @@ -19,7 +19,7 @@ from setuptools.command.build_ext import build_ext # This needs to happen before get_extensions() is called # Usage: python setup.py install --no-cuda # or: pip install . --no-cuda -BUILD_NO_CUDA = False +BUILD_NO_CUDA = os.getenv("COMFY_KITCHEN_NO_CUDA", "0") == "1" if "--no-cuda" in sys.argv: BUILD_NO_CUDA = True sys.argv.remove("--no-cuda") # Remove so setuptools doesn't complain