diff --git a/models/cv/classification/deit_tiny/ixrt/README.md b/models/cv/classification/deit_tiny/ixrt/README.md
index 7a58584d058a99a036b9b201b0821cb03de7015b..1ceea7bfbbb8bb888cc0cb2a14bc5de4773962fa 100644
--- a/models/cv/classification/deit_tiny/ixrt/README.md
+++ b/models/cv/classification/deit_tiny/ixrt/README.md
@@ -34,10 +34,11 @@ Dataset: <https://www.image-net.org/download.php> to download the validation dat
 
 ```bash
 # git clone mmpretrain
-git clone -b v0.24.0 https://github.com/open-mmlab/mmpretrain.git
+git clone --depth 1 -b v1.1.0 https://github.com/open-mmlab/mmpretrain.git
+(cd mmpretrain/ && python3 setup.py develop)
 
 # export onnx model
-python3 export.py --cfg mmpretrain/configs/deit/deit-tiny_pt-4xb256_in1k.py --weight deit-tiny_pt-4xb256_in1k_20220218-13b382a0.pth --output deit_tiny.onnx
+python3 export.py --cfg mmpretrain/configs/deit/deit-tiny_4xb256_in1k.py --weight deit-tiny_pt-4xb256_in1k_20220218-13b382a0.pth --output deit_tiny.onnx
 
 # Use onnxsim optimize onnx model
 onnxsim deit_tiny.onnx deit_tiny_opt.onnx
diff --git a/models/cv/classification/deit_tiny/ixrt/export.py b/models/cv/classification/deit_tiny/ixrt/export.py
index 0078670ffbf4bbcce3358d4a2cedc42ce61176f5..b2f2f14217edcdda5ae6d4c2fa747be47b4e4944 100644
--- a/models/cv/classification/deit_tiny/ixrt/export.py
+++ b/models/cv/classification/deit_tiny/ixrt/export.py
@@ -15,7 +15,7 @@
 import argparse
 
 import torch
-from mmcls.apis import init_model
+from mmpretrain.apis import init_model
 
 class Model(torch.nn.Module):
     def __init__(self, config_file, checkpoint_file):