minor cleanup

megatron/arguments.py

@@ -620,8 +620,6 @@ def _add_realm_args(parser):
 def _add_vit_args(parser):
     group = parser.add_argument_group(title="vit")
 
-    group.add_argument('--vit-load', type=str, default=None,
-                       help='Director containing a VitModel checkpoint')
     group.add_argument('--num-classes', type=int, default=1000,
                        help='num of classes in vision classificaiton task')
     group.add_argument('--img-dim', type=int, default=224,

megatron/checkpointing.py

@@ -59,7 +59,7 @@ def check_checkpoint_args(checkpoint_args):
     _compare('hidden_size')
     _compare('num_attention_heads')
     _compare('max_position_embeddings')
-    if args.vit_load is None:
+    if args.vocab_file:
         _compare('make_vocab_size_divisible_by')
         _compare('padded_vocab_size')
         _compare('tokenizer_type')