Update src/diffusers/modeling_flax_utils.py

Co-authored-by: Suraj Patil <[email protected]>

Author: Mishig Davaadorj

src/diffusers/modeling_flax_utils.py

@@ -184,7 +184,7 @@ def to_fp16(self, params: Union[Dict, FrozenDict], mask: Any = None):
         return self._cast_floating_to(params, jnp.float16, mask)
 
     def init_weights(self, rng: jax.random.PRNGKey) -> Dict:
-        raise NotImplementedError(f"init method has to be implemented for {self}")
+        raise NotImplementedError(f"init_weights method has to be implemented for {self}")
 
     @classmethod
     def from_pretrained(