src/diffusers/pipelines/stable_diffusion/safety_checker_flax.py · c6629e6f111114e4432678ddd4beb83571dc25b9 · chenpangpang / diffusers · GitLab

Find file Blame History Permalink

[flax safety checker] Use `FlaxPreTrainedModel` for saving/loading (#591) · c6629e6f
Suraj Patil authored Sep 20, 2022
```
* use FlaxPreTrainedModel for flax safety module

* fix name

* fix one more

* Apply suggestions from code review
```
c6629e6f

safety_checker_flax.py 5.84 KB