↓ 2 callersMethodget_latents(self, seed=0, shape=(4, 4, 64, 64), fp16=False)
diffusers/tests/models/test_models_unet_2d_flax.py:27
↓ 2 callersFunctionlog_validation(pipeline, pipeline_params, controlnet_params, tokenizer, args, rng, weight_dtype)
diffusers/examples/controlnet/train_controlnet_flax.py:79