ComfyUI/comfy_api/torch_helpers/__init__.py

from .torch_compile import set_torch_compile_wrapper

__all__ = [
    "set_torch_compile_wrapper",
]
Make torch.compile LoRA/key-compatible (#8213) * Make torch compile node use wrapper instead of object_patch for the entire diffusion_models object, allowing key assotiations on diffusion_models to not break (loras, getting attributes, etc.) * Moved torch compile code into comfy_api so it can be used by custom nodes with a degree of confidence * Refactor set_torch_compile_wrapper to support a list of keys instead of just diffusion_model, as well as additional torch.compile args * remove unused import * Moved torch compile kwargs to be stored in model_options instead of attachments; attachments are more intended for things to be 'persisted', AKA not deepcopied * Add some comments * Remove random line of code, not sure how it got there 2025-05-21 01:56:56 -07:00			`from .torch_compile import set_torch_compile_wrapper`

			`__all__ = [`
			`"set_torch_compile_wrapper",`
			`]`