fix

huggingface · Oct 15, 2024 · 6183a8b · 6183a8b
1 parent 23a02bd
commit 6183a8b
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/optimum/onnxruntime/modeling_decoder.py b/optimum/onnxruntime/modeling_decoder.py
@@ -161,7 +161,6 @@ def __init__(
                     setattr(self.generation_config, param_name, param_value)
                     setattr(self.config, param_name, None)
 
-        self.model_path = Path(model._model_path)
         self.use_merged = "use_cache_branch" in self.input_names
         self.model_type = self.config.model_type
 

diff --git a/optimum/onnxruntime/modeling_ort.py b/optimum/onnxruntime/modeling_ort.py
@@ -247,6 +247,8 @@ def __init__(
     ):
         super().__init__(model, config)
 
+        self.model_path = Path(model._model_path)
+
         self.input_names = {input_key.name: idx for idx, input_key in enumerate(model.get_inputs())}
         self.input_dtypes = {input_key.name: input_key.type for input_key in model.get_inputs()}