feat(model): set default config for legacy models

dalle_mini/model.py

@@ -46,6 +46,11 @@ class CustomFlaxBartForConditionalGenerationModule(
     FlaxBartForConditionalGenerationModule
 ):
     def setup(self):
+        # set default config
+        self.config.normalize_text = getattr(self.config, "normalize_text", False)
+        self.config.image_length = getattr(self.config, "image_length", 256)
+        self.config.image_vocab_size = getattr(self.config, "image_vocab_size", 16384)
+
         self.model = CustomFlaxBartModule(config=self.config, dtype=self.dtype)
         self.lm_head = nn.Dense(
             self.config.image_vocab_size + 1,  # encoded image token space + 1 for bos