Mirrors/modelscope
1
0
Fork 0
mirror of https://github.com/modelscope/modelscope.git synced 2026-02-24 20:19:51 +01:00
Code Issues Packages Projects Releases Wiki Activity

remove megatron_util initlize twice

Browse Source
This commit is contained in:
mulin.lyh
2023-11-29 23:25:19 +08:00
parent dcf4b39830
commit 602fab39d4
1 changed files with 0 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
modelscope/utils/pre_compile.py
View File

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