There is already a condition check within the function. This outer on…

…e is not necessary Co-authored-by: NanoCode012 <kevinvong@rocketmail.com>
axolotl-ai-cloud · Aug 16, 2024 · 7f59dac · 7f59dac
1 parent 3b22fb8
commit 7f59dac
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/src/axolotl/utils/trainer.py b/src/axolotl/utils/trainer.py
@@ -442,8 +442,7 @@ def prepare_optim_env(cfg):
             stage = deepspeed_config.get("zero_optimization", {}).get("stage", None)
         setup_deepspeed_env(cfg, stage=stage)
 
-    if cfg.torch_compile:
-        setup_torch_compile_env(cfg)
+    setup_torch_compile_env(cfg)
 
     if (cfg.bf16 == "auto" and is_torch_bf16_gpu_available()) or cfg.bf16 is True:
         os.environ["ACCELERATE_MIXED_PRECISION"] = "bf16"