Spaces:

tom-doerr
/

logo_generator

Runtime error

boris commited on Feb 7, 2022

Commit

68cc185

1 Parent(s): 44b7c3e

fix: typo

Files changed (1) hide show

src/dalle_mini/model/modeling.py CHANGED Viewed

@@ -337,7 +337,7 @@ class FlaxBartPreTrainedModel(FlaxBartPreTrainedModel):
             # init weights on CPU
             init_fn = jax.jit(self.init_weights, static_argnums=(1,), backend="cpu")
         else:
-            init_fn = self.init_weigths
         # randomly initialized parameters
         random_params = self.init_weights(self.key, input_shape)

             # init weights on CPU
             init_fn = jax.jit(self.init_weights, static_argnums=(1,), backend="cpu")
         else:
+            init_fn = self.init_weights
         # randomly initialized parameters
         random_params = self.init_weights(self.key, input_shape)