Open-Sora/opensora
Gao, Ruiyuan df5668cdf1 fix bug at mha, MaskGenerator; improve ckpt_utils.py (#609)
* fix bug at mha in blocks.py

* fix bug in MaskGenerator

* align logging style in ckpt_utils.py
2025-02-20 16:40:47 +08:00
..
acceleration added timer for benchmarking (#114) 2024-05-29 14:43:15 +08:00
datasets [feature] support async ckpt & pin memory cache (#760) 2024-12-20 10:30:49 +08:00
models fix bug at mha, MaskGenerator; improve ckpt_utils.py (#609) 2025-02-20 16:40:47 +08:00
schedulers lint: unify setting in pyproject.toml (#583) 2024-07-12 14:50:13 +08:00
utils fix bug at mha, MaskGenerator; improve ckpt_utils.py (#609) 2025-02-20 16:40:47 +08:00
__init__.py [fix] vae dtype 2024-05-16 08:50:24 +00:00
registry.py [wip] feat training 2024-05-21 05:45:06 +00:00