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():