From b5bc651883654e102c938378a17d055be259833d Mon Sep 17 00:00:00 2001 From: abdullah-ibm Date: Wed, 30 Oct 2024 11:54:15 +0300 Subject: [PATCH] typo fix --- 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 1b017c82..3a7e1225 100644 --- 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(