From 104cb93d507e5dafbc778ce70c6772481e01852d Mon Sep 17 00:00:00 2001
From: "mulin.lyh" <mulin.lyh@taobao.com>
Date: Fri, 14 Jul 2023 19:11:49 +0800
Subject: [PATCH] [to #50841229]feat: precompile pai-easycv in image Link:
 https://code.alibaba-inc.com/Ali-MaaS/MaaS-lib/codereview/13225347 * [to
 #50841229]feat: precompile pai-easycv in image

---
 modelscope/utils/pre_compile.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/modelscope/utils/pre_compile.py b/modelscope/utils/pre_compile.py
index 65bde30a..ab9c2518 100644
--- a/modelscope/utils/pre_compile.py
+++ b/modelscope/utils/pre_compile.py
@@ -20,6 +20,8 @@ def pre_compile_all():
     if torch.cuda.is_available():  # extension require cuda.
         pre_compile_megatron_util()
 
+    # pre compile pai-easycv
+    from easycv.thirdparty.deformable_attention.functions import ms_deform_attn_func
     # extension for all platform.