tocmo0nlord/axolotl
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

chore: refactor if condition

Browse Source
This commit is contained in:
NanoCode012
2025-05-22 18:25:45 +07:00
parent 2e2f42918d
commit 58842ded9c
1 changed files with 6 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/axolotl/core/trainer_builder/base.py
View File

@@ -452,8 +452,10 @@ class TrainerBuilderBase(abc.ABC):
optim_args = self.cfg.optim_args optim_args = self.cfg.optim_args
training_args_kwargs["optim_args"] = optim_args training_args_kwargs["optim_args"] = optim_args
if self.cfg.optimizer == "adamw_anyprecision": if (
if Path(self.cfg.torchdistx_path).exists(): self.cfg.optimizer == "adamw_anyprecision"
and Path(self.cfg.torchdistx_path).exists()
):
sys.path.append(self.cfg.torchdistx_path) sys.path.append(self.cfg.torchdistx_path)
importlib.import_module("torchdistx") importlib.import_module("torchdistx")