Fixed naming error

load_models.py

@@ -28,7 +28,7 @@ def get_model(args):
         if args.model == "roberta":
             num_labels = 1 if args.dataset == "glue_stsb" else 2  # TODO might be wrong
             config = AutoConfig.from_pretrained(
-                args.model_name_or_path,
+                "roberta-base",
                 num_labels=num_labels,
                 finetuning_task=arg_map[args.dataset][1],
             )