typo fix

instructlab · Oct 30, 2024 · b5bc651 · b5bc651
1 parent 2615349
commit b5bc651
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
@@ -533,7 +533,7 @@ def main(args):
     if args.distributed_training_framework == 'deepspeed' and not FusedAdam:
         raise ImportError("DeepSpeed was selected but we cannot import the `FusedAdam` optimizer")
 
-    if args.distributed_training_framework == 'deeppeed' and args.cpu_offload_optimizer and not DeepSpeedCPUAdam:
+    if args.distributed_training_framework == 'deepspeed' and args.cpu_offload_optimizer and not DeepSpeedCPUAdam:
         raise ImportError("DeepSpeed was selected and CPU offloading was requested, but DeepSpeedCPUAdam could not be imported. This likely means you need to build DeepSpeed with the CPU adam flags.") 
 
     metric_logger = AsyncStructuredLogger(