ComfyUI/comfy/ldm/hydit
comfyanonymous a5f4292f9f
Basic hunyuan dit implementation. (#4102)
* Let tokenizers return weights to be stored in the saved checkpoint.

* Basic hunyuan dit implementation.

* Fix some resolutions not working.

* Support hydit checkpoint save.

* Init with right dtype.

* Switch to optimized attention in pooler.

* Fix black images on hunyuan dit.
2024-07-25 18:21:08 -04:00
..
attn_layers.py Basic hunyuan dit implementation. (#4102) 2024-07-25 18:21:08 -04:00
models.py Basic hunyuan dit implementation. (#4102) 2024-07-25 18:21:08 -04:00
poolers.py Basic hunyuan dit implementation. (#4102) 2024-07-25 18:21:08 -04:00
posemb_layers.py Basic hunyuan dit implementation. (#4102) 2024-07-25 18:21:08 -04:00