diff --git a/deeplink_ext/patch_internlm.py b/deeplink_ext/patch_internlm.py index 8651c6e5..749b4b75 100644 --- a/deeplink_ext/patch_internlm.py +++ b/deeplink_ext/patch_internlm.py @@ -87,7 +87,7 @@ def backward(ctx, dqkv: torch.Tensor, *args, **kwargs): _find_or_mock_module("flash_attn_cuda") _find_flash_attn() _patch_flash_attn() - _patch_ops() + # _patch_ops() print("[deeplink_ext] patched diopi implementation of internlm\n", end="")