From ea3ec9c927931ef0dc33a952a33a982ae1e562a2 Mon Sep 17 00:00:00 2001 From: Oleg S <97077423+RobotSail@users.noreply.github.com> Date: Tue, 12 Nov 2024 14:01:44 +0000 Subject: [PATCH] fix: incorrect enum selection Signed-off-by: Oleg S <97077423+RobotSail@users.noreply.github.com> --- src/instructlab/training/main_ds.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/instructlab/training/main_ds.py b/src/instructlab/training/main_ds.py index 63c92784..4c04da0f 100644 --- a/src/instructlab/training/main_ds.py +++ b/src/instructlab/training/main_ds.py @@ -775,7 +775,7 @@ def run_training(torch_args: TorchrunArgs, train_args: TrainingArgs) -> None: ) # deepspeed options - if train_args.distributed_backend == DistributedBackend.DeepSpeed: + if train_args.distributed_backend == DistributedBackend.DEEPSPEED: if not FusedAdam: raise ImportError( "DeepSpeed was selected as the distributed backend, but FusedAdam could not be imported. Please double-check that DeepSpeed is installed correctly"