Revert #1819 + make style

huggingface · Mar 6, 2025 · 43852a6 · 43852a6
1 parent daeeda9
commit 43852a6
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/examples/text-generation/run_lm_eval.py b/examples/text-generation/run_lm_eval.py
@@ -27,17 +27,17 @@
 import psutil
 import torch
 import torch.nn.functional as F
+import transformers
 from lm_eval import evaluator, utils
 from lm_eval.models.huggingface import HFLM, TemplateLM
 
 # Local imports
 from run_generation import setup_parser
-import transformers
 from transformers import AutoModelForCausalLM, AutoTokenizer
 from transformers.generation import GenerationConfig
-from optimum.habana.transformers.generation import GaudiGenerationConfig
 from utils import finalize_quantization, initialize_model, save_model
 
+from optimum.habana.transformers.generation import GaudiGenerationConfig
 from optimum.habana.utils import get_hpu_memory_stats
 
 

diff --git a/optimum/habana/transformers/modeling_utils.py b/optimum/habana/transformers/modeling_utils.py
@@ -376,7 +376,6 @@ def adapt_transformers_to_gaudi():
         GaudiGenerationMixin._prepare_cache_for_generation
     )
     transformers.generation.GenerationConfig = GaudiGenerationConfig
-    transformers.GenerationConfig = GaudiGenerationConfig
     transformers.generation.configuration_utils.GenerationConfig = GaudiGenerationConfig
     transformers.modeling_utils.GenerationConfig = GaudiGenerationConfig
     transformers.generation.MaxLengthCriteria.__call__ = gaudi_MaxLengthCriteria_call