We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 51347f9 commit 5cb1e0cCopy full SHA for 5cb1e0c
comfy_extras/nodes_torch_compile.py
@@ -2,6 +2,8 @@
2
from comfy_api.latest import ComfyExtension, io
3
from comfy_api.torch_helpers import set_torch_compile_wrapper
4
5
+def skip_torch_compile_dict(guard_entries):
6
+ return [("transformer_options" not in entry.name) for entry in guard_entries]
7
8
class TorchCompileModel(io.ComfyNode):
9
@classmethod
@@ -23,7 +25,7 @@ def define_schema(cls) -> io.Schema:
23
25
24
26
def execute(cls, model, backend) -> io.NodeOutput:
27
m = model.clone()
- set_torch_compile_wrapper(model=m, backend=backend)
28
+ set_torch_compile_wrapper(model=m, backend=backend, options={"guard_filter_fn": skip_torch_compile_dict})
29
return io.NodeOutput(m)
30
31
0 commit comments