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