Hongxin Liu 332d9fc9c9 [feature] make timer optional and make reduce bucket size configurable (#549) ... * [feature] make reduce bucket size configurable * [feature] make timer optional		2024-06-27 13:37:54 +08:00
..
acceleration	added timer for benchmarking (#114)	2024-05-29 14:43:15 +08:00
datasets	handle av error	2024-06-22 13:26:55 +00:00
models	fix SeqParallelMultiHeadCrossAttention for consistent results in distributed mode (#510)	2024-06-24 17:07:49 +08:00
schedulers	Force fp16 input to fp32 to avoid nan output in timestep_transform	2024-06-21 11:15:39 +03:00
utils	[feature] make timer optional and make reduce bucket size configurable (#549)	2024-06-27 13:37:54 +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