This commit is contained in:
kijai 2024-10-25 18:36:03 +03:00
parent 25eeab3c4c
commit ca5dfdf79c

View File

@ -198,7 +198,7 @@ class MochiTorchCompileSettings:
def INPUT_TYPES(s): def INPUT_TYPES(s):
return { return {
"required": { "required": {
"backend": (["inductor","cudagraph"], {"default": "inductor"}), "backend": (["inductor","cudagraphs"], {"default": "inductor"}),
"fullgraph": ("BOOLEAN", {"default": False, "tooltip": "Enable full graph mode"}), "fullgraph": ("BOOLEAN", {"default": False, "tooltip": "Enable full graph mode"}),
"mode": (["default", "max-autotune", "max-autotune-no-cudagraphs", "reduce-overhead"], {"default": "default"}), "mode": (["default", "max-autotune", "max-autotune-no-cudagraphs", "reduce-overhead"], {"default": "default"}),
"compile_dit": ("BOOLEAN", {"default": True, "tooltip": "Compiles all transformer blocks"}), "compile_dit": ("BOOLEAN", {"default": True, "tooltip": "Compiles all transformer blocks"}),