fix(decoder): use revision when fetching generation_config

huggingface · Apr 25, 2024 · a59f629 · a59f629
1 parent e364105
commit a59f629
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/optimum/neuron/modeling_decoder.py b/optimum/neuron/modeling_decoder.py
@@ -362,7 +362,7 @@ def _export(
         # Try to reload the generation config (if any)
         generation_config = None
         try:
-            generation_config = GenerationConfig.from_pretrained(model_id)
+            generation_config = GenerationConfig.from_pretrained(model_id, revision=revision)
         except OSError:
             pass
 
@@ -419,7 +419,7 @@ def _from_pretrained(
         # Try to reload the generation config (if any)
         generation_config = None
         try:
-            generation_config = GenerationConfig.from_pretrained(model_id)
+            generation_config = GenerationConfig.from_pretrained(model_id, revision=revision)
         except OSError:
             pass