Mirrors/modelscope
1
0
Fork 0
mirror of https://github.com/modelscope/modelscope.git synced 2026-02-25 04:30:48 +01:00
Code Issues Packages Projects Releases Wiki Activity

[to #50841229]feat: precompile pai-easycv in image

Browse Source 
Link: https://code.alibaba-inc.com/Ali-MaaS/MaaS-lib/codereview/13225347
* [to #50841229]feat: precompile pai-easycv in image
This commit is contained in:
mulin.lyh
2023-07-14 19:11:49 +08:00
parent ceac129c6b
commit 104cb93d50
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
modelscope/utils/pre_compile.py
View File

@@ -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.