fix

huggingface · Sep 7, 2024 · 80c63d0 · 80c63d0
1 parent f0e3f2b
commit 80c63d0
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/optimum/onnxruntime/base.py b/optimum/onnxruntime/base.py
@@ -139,8 +139,8 @@ def __init__(
         super().__init__(session, parent_model)
 
         config = (
-            self.parent_model.config.encoder
-            if hasattr(self.parent_model.config, "encoder")
+            self.parent_model.config.decoder
+            if hasattr(self.parent_model.config, "decoder")
             else self.parent_model.config
         )