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 | ||