diffusers

Move safety detection to model call in Flax safety checker

#1023

Merged