Update modeling.py

modeling.py

@@ -20,10 +20,10 @@ class DigitDiffusionOutput(ModelOutput):
 
 
 class DigitDiffusionModel(PreTrainedModel):
-
     config_class = DigitDiffusionConfig
     base_model_prefix = "unet"
     main_input_name = "noisy_images"
+    all_tied_weights_keys = {}
 
     def __init__(self, config: DigitDiffusionConfig) -> None:
         super().__init__(config)
@@ -45,6 +45,8 @@ class DigitDiffusionModel(PreTrainedModel):
             class_embed_type=config.class_embed_type,
         )
 
+        self.post_init()
+
     def _init_weights(self, module):
         # Diffusers initializes the UNet internally, so there is nothing extra
         # to initialize here.