Reformatting, remove TODO.

app.py

@@ -181,7 +181,8 @@ def create_app():
             label='Prompts (press Shift-ENTER to add a prompt)')
         with gr.Row():
 
-          family = gr.Dropdown(value='lit', choices=list(MODEL_MAP), label='Model family')
+          family = gr.Dropdown(
+              value='lit', choices=list(MODEL_MAP), label='Model family')
 
           make_variant = functools.partial(gr.Dropdown, label='Variant')
           variant = make_variant(list(MODEL_MAP['lit']), value='B/16')
@@ -221,10 +222,10 @@ def create_app():
             ]
 
           gr.on(
-            [family.change, variant.change, res.change],
-            update_inputs,
-            [family, variant, res],
-            [variant, res, bias],
+              [family.change, variant.change, res.change],
+              update_inputs,
+              [family, variant, res],
+              [variant, res, bias],
           )
 
           # (end of code for reactive UI)

big_vision_contrastive_models.py

@@ -211,7 +211,7 @@ def load_model(config, check_params=False):
   assert isinstance(config, ContrastiveModelConfig), type(config)
 
   cfg = ml_collections.ConfigDict()
-  cfg.image_model = 'vit'  # TODO(lbeyer): remove later, default
+  cfg.image_model = 'vit'
   if config.family == ContrastiveModelFamily.LIT:
     cfg.text_model = 'proj.flaxformer.bert'
     cfg.image = dict(