Zheng Zangwei (Alex Zheng) 9b668e1c4e Merge pull request #523 from BurkeHulk/hotfix/fp16_nan_output ... Force fp16 input to fp32 to avoid nan output in timestep_transform		2024-06-21 18:01:17 +08:00
..
acceleration	added timer for benchmarking (#114)	2024-05-29 14:43:15 +08:00
datasets	fix ar keys (#500)	2024-06-20 17:51:24 +08:00
models	config for local load	2024-06-20 10:23:38 +00:00
schedulers	Force fp16 input to fp32 to avoid nan output in timestep_transform	2024-06-21 11:15:39 +03:00
utils	reformat and update docs	2024-06-17 15:37:23 +00: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