From 11c2ee5f7231b5bf0e2651c6a18992f7793ae750 Mon Sep 17 00:00:00 2001 From: github-actions Date: Mon, 18 Sep 2023 08:35:07 +0000 Subject: [PATCH] [format] applied code formatting on changed files in pull request 4743 --- .../language/gpt/experiments/pipeline_parallel/train_gpt_pp.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/language/gpt/experiments/pipeline_parallel/train_gpt_pp.py b/examples/language/gpt/experiments/pipeline_parallel/train_gpt_pp.py index 30d6aab4f12f..749243e57836 100644 --- a/examples/language/gpt/experiments/pipeline_parallel/train_gpt_pp.py +++ b/examples/language/gpt/experiments/pipeline_parallel/train_gpt_pp.py @@ -3,6 +3,7 @@ from functools import partial import torch +from model_zoo import model_builder from torch import nn from tqdm import tqdm @@ -18,7 +19,6 @@ from colossalai.legacy.pipeline.rpc.utils import rpc_run from colossalai.logging import disable_existing_loggers, get_dist_logger from colossalai.nn.optimizer import HybridAdam -from model_zoo import model_builder def parse_args():