a5f4292f9f
* 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. |
||
---|---|---|
.. | ||
attn_layers.py | ||
models.py | ||
poolers.py | ||
posemb_layers.py |