From 10448e67efde82f2a5793eb0133ea4ee272d2378 Mon Sep 17 00:00:00 2001
From: doctorpangloss <@hiddenswitch.com>
Date: Fri, 12 Sep 2025 14:58:26 -0700
Subject: [PATCH] Tweak torch compile

---
 comfy_api/torch_helpers/torch_compile.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/comfy_api/torch_helpers/torch_compile.py b/comfy_api/torch_helpers/torch_compile.py
index 721fd43aa..4827ff6d2 100644
--- a/comfy_api/torch_helpers/torch_compile.py
+++ b/comfy_api/torch_helpers/torch_compile.py
@@ -16,7 +16,7 @@ COMPILE_KEY = "torch.compile"
 TORCH_COMPILE_KWARGS = "torch_compile_kwargs"
 
 
-def apply_torch_compile_factory(compiled_module_dict: dict[str, Callable], model_patcher: Optional[ModelPatcher] = None) -> Callable:
+def apply_torch_compile_factory(compiled_module_dict: dict[str, Callable]) -> Callable:
     '''
     Create a wrapper that will refer to the compiled_diffusion_model.
     '''
@@ -67,7 +67,6 @@ def set_torch_compile_wrapper(model: ModelPatcher, backend: str, options: Option
     # add torch.compile wrapper
     wrapper_func = apply_torch_compile_factory(
         compiled_module_dict=compiled_modules,
-        model_patcher=model,
     )
     # store wrapper to run on BaseModel's apply_model function
     model.add_wrapper_with_key(WrappersMP.APPLY_MODEL, COMPILE_KEY, wrapper_func)