Merge pull request #331 from RobotSail/fix-training-backend-selection

fix: incorrect enum selection
instructlab · Nov 12, 2024 · 1a24cce · 1a24cce
2 parents b812604 + ea3ec9c
commit 1a24cce
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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"