Fix crash if gaudi_config is not passed to GaudiTrainer (#613)

huggingface · Dec 29, 2023 · 0152688 · 0152688
1 parent a5d7dec
commit 0152688
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/optimum/habana/transformers/trainer.py b/optimum/habana/transformers/trainer.py
@@ -214,7 +214,7 @@ def __init__(
                     )
 
             if self.use_hpu_amp and "LOWER_LIST" not in os.environ:
-                gaudi_config.declare_autocast_bf16_fp32_ops()
+                self.gaudi_config.declare_autocast_bf16_fp32_ops()
 
             if self.args.use_lazy_mode:
                 try: