fix typo

0e9a705a · Geewook Kim · GitHub · 0c962553 · 0e9a705a
Unverified Commit 0e9a705a authored Jul 29, 2022 by Geewook Kim Committed by GitHub Jul 29, 2022
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

test.py test.py +2 -2

No files found.
--- a/test.py
+++ b/test.py
@@ -19,7 +19,7 @@ from donut import DonutModel, JSONParseEvaluator, load_json, save_json


 def test(args):
-    pretrained_model = DonutModel.from_pretrained(args.pretrained_path)
+    pretrained_model = DonutModel.from_pretrained(args.pretrained_model_name_or_path)

    if torch.cuda.is_available():
        pretrained_model.half()
@@ -74,7 +74,7 @@ def test(args):

 if __name__ == "__main__":
    parser = argparse.ArgumentParser()
-    parser.add_argument("--pretrained_path", type=str)
+    parser.add_argument("--pretrained_model_name_or_path", type=str)
    parser.add_argument("--dataset_name_or_path", type=str)
    parser.add_argument("--split", type=str, default="test")
    parser.add_argument("--task_name", type=str, default=None)