ops!: align to refactored deeplink_ext

see DeepLink-org/DeepLinkExt#36
DeepLink-org · Jan 24, 2024 · 2291383 · 2291383
1 parent ff3baf3
commit 2291383
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 81 deletions.
diff --git a/patch_deeplinkext.py b/patch_deeplinkext.py
diff --git a/train.py b/train.py
@@ -6,11 +6,10 @@
 import traceback
 from functools import partial
 
-import torch_dipu
 import torch
 import torch.distributed as dist
 
-import patch_deeplinkext
+import deeplink_ext.patch_internlm
 import internlm
 from internlm.core.context import ParallelMode
 from internlm.core.context import global_context as gpc