diffusers
470f51cd - feat: add `act_fn` param to `OutValueFunctionBlock` (#3994)

Commit
2 years ago
feat: add `act_fn` param to `OutValueFunctionBlock` (#3994) * feat: add act_fn param to OutValueFunctionBlock * feat: update unet1d tests to not use mish * feat: add `mish` as the default activation function Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com> * feat: drop mish tests from unet1d --------- Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com>
Parents
Loading