Add name_or_path for donut generation (#1345)

add name_or_path
huggingface · Sep 4, 2023 · 0c399f9 · 0c399f9
1 parent 659cf02
commit 0c399f9
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 7 deletions.
diff --git a/optimum/onnxruntime/modeling_seq2seq.py b/optimum/onnxruntime/modeling_seq2seq.py
@@ -607,6 +607,7 @@ def show_deprecated_argument(arg_name):
             preprocessors=preprocessors,
         )
         self.config = config
+        self.name_or_path = config.name_or_path
 
         self.onnx_paths = onnx_paths
         self.use_cache = use_cache

diff --git a/tests/benchmark/benchmark_bettertransformer.py b/tests/benchmark/benchmark_bettertransformer.py
@@ -1,13 +1,14 @@
 import argparse
 
+import numpy as np
+import pandas as pd
 import torch
 from tqdm import tqdm
 from transformers import AutoModel, AutoModelForCausalLM, AutoModelForSeq2SeqLM, AutoTokenizer, GenerationConfig
 
 from optimum.bettertransformer import BetterTransformer
 from optimum.exporters import TasksManager
-import numpy as np
-import pandas as pd
+
 
 def get_parser():
     parser = argparse.ArgumentParser()
@@ -59,11 +60,7 @@ def get_parser():
         "--use-mask",
         action="store_true",
     )
-    parser.add_argument(
-        "--is_decoder",
-        action="store_true",
-        help="Benchmark the generate method."
-    )
+    parser.add_argument("--is_decoder", action="store_true", help="Benchmark the generate method.")
     parser.add_argument(
         "--sweep",
         action="store_true",