ComfyUI/comfy_extras
Jedrzej Kosinski 65da29aaa9
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 04:56:56 -04:00
..
2025-05-13 10:50:32 -07:00
2024-05-27 19:08:27 -04:00
2025-03-26 05:09:52 -04:00
2025-04-30 20:57:30 -04:00
2025-05-06 04:23:00 -04:00
2024-11-10 00:10:45 -05:00
2024-05-27 19:08:27 -04:00
2024-03-18 16:54:13 -04:00
2025-05-01 19:32:04 -04:00