diffusers
9d0d0709 - EMA: fix `state_dict()` and `load_state_dict()` & add `cur_decay_value` (#2146)

Commit
3 years ago
EMA: fix `state_dict()` and `load_state_dict()` & add `cur_decay_value` (#2146) * EMA: fix `state_dict()` & add `cur_decay_value` * EMA: fix a bug in `load_state_dict()` 'float' object (`state_dict["power"]`) has no attribute 'get'. * del train_unconditional_ort.py
Author
Parents
Loading