fix: typo

borisdayma · Feb 7, 2022 · 68cc185 · 68cc185
1 parent 44b7c3e
commit 68cc185
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/dalle_mini/model/modeling.py b/src/dalle_mini/model/modeling.py
@@ -337,7 +337,7 @@ def __init__(
             # init weights on CPU
             init_fn = jax.jit(self.init_weights, static_argnums=(1,), backend="cpu")
         else:
-            init_fn = self.init_weigths
+            init_fn = self.init_weights
 
         # randomly initialized parameters
         random_params = self.init_weights(self.key, input_shape)