63c68d979a
* improve comments for sde_ve scheduler, init tests * more comments, tweaking pipelines * timesteps --> num_training_timesteps, some comments * merge cpu test, add m1 data * fix scheduler tests with num_train_timesteps * make np compatible, add tests for sde ve * minor default variable fixes * make style and fix-copies Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com> |
||
---|---|---|
.. | ||
__init__.py | ||
test_layers_utils.py | ||
test_modeling_utils.py | ||
test_scheduler.py |