Fix some small things

configs/models/cc12m_256x256.yaml

@@ -37,7 +37,7 @@ unet_config:
   conditioning_feature_dim: -1
   conditioning_feature_proj_dim: -1
   freeze_inner_unet: false
-  initialize_inner_with_pretrained: None
+  initialize_inner_with_pretrained: null
   inner_config:
     attention_levels: [1, 2]
     conditioning_feature_dim: -1

ml_mdm/clis/generate_sample.py

@@ -144,7 +144,7 @@ def generate(
         prompt = input_template.format(prompt=prompt)
     if len(negative_template) > 0:
         negative_prompt = negative_prompt + negative_template
-    print(f"Postive: {prompt} / Negative: {negative_prompt}")
+    print(f"Positive: {prompt} / Negative: {negative_prompt}")
 
     if not os.path.exists(ckpt_name):
         logging.info(f"Did not generate because {ckpt_name} does not exist")
@@ -484,7 +484,7 @@ def main(args):
 
             with gr.Column(scale=2):
                 with gr.Accordion(
-                    "Addditional outputs", open=False, elem_id="output-accordion"
+                    "Additional outputs", open=False, elem_id="output-accordion"
                 ):
                     with gr.Row(equal_height=True):
                         output_text = gr.Textbox(value=None, label="System output")

ml_mdm/models/nested_unet.py

@@ -76,6 +76,9 @@ class Nested4UNetConfig(Nested3UNetConfig):
 
 
 def download(vision_model_path):
+    from ml_mdm.utils import fix_old_checkpoints
+    fix_old_checkpoints.mimic_old_modules()
+
     import os
 
     from distributed import get_local_rank