ComfyUI/comfy/ldm/modules
mengqin 0357ed7ec4
Add support for sage attention 3 in comfyui, enable via new cli arg (#11026)
* Add support for sage attention 3 in comfyui, enable via new cli arg
--use-sage-attiention3

* Fix some bugs found in PR review. The N dimension at which Sage
Attention 3 takes effect is reduced to 1024 (although the improvement is
not significant at this scale).

* Remove the Sage Attention3 switch, but retain the attention function
registration.

* Fix a ruff check issue in attention.py
2025-12-30 22:53:52 -05:00
..
diffusionmodules Implement temporal rolling VAE (Major VRAM reductions in Hunyuan and Kandinsky) (#10995) 2025-12-02 22:49:29 -05:00
distributions Small optimizations. 2024-12-18 18:23:28 -05:00
encoders Make unclip more deterministic. 2024-01-14 17:28:31 -05:00
attention.py Add support for sage attention 3 in comfyui, enable via new cli arg (#11026) 2025-12-30 22:53:52 -05:00
ema.py Initial commit. 2023-01-16 22:37:14 -05:00
sub_quadratic_attention.py Fix pytorch warning. (#8593) 2025-06-19 11:04:52 -04:00
temporal_ae.py Basic Hunyuan Video model support. 2024-12-16 19:35:40 -05:00