From 88ad7915b069bd592260f13d971bd43e13cf32a8 Mon Sep 17 00:00:00 2001 From: Seth Ockerman Date: Wed, 13 Mar 2024 21:47:11 -0500 Subject: [PATCH] replaced outdated import path of get_forward_backward_func with updated import path --- tasks/eval_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tasks/eval_utils.py b/tasks/eval_utils.py index 98d1bfb2ed..3078055f2d 100644 --- a/tasks/eval_utils.py +++ b/tasks/eval_utils.py @@ -11,7 +11,7 @@ from megatron import get_args from megatron import print_rank_last, is_last_rank from megatron.core import mpu -from megatron.schedules import get_forward_backward_func +from megatron.core.pipeline_parallel import get_forward_backward_func from tasks.finetune_utils import build_data_loader from tasks.finetune_utils import process_batch