diffusers

e3095c5f - Fix invocation of some slow Flax tests (#3058)

Fix invocation of some slow Flax tests (#3058) * Fix invocation of some slow tests. We use __call__ rather than pmapping the generation function ourselves because the number of static arguments is different now. * style