Skip to content

Commit aeebd59

Browse files
fix multi node (#1554)
1 parent 3fad431 commit aeebd59

File tree

1 file changed

+10
-0
lines changed

1 file changed

+10
-0
lines changed

swift/llm/utils/model.py

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -27,6 +27,7 @@
2727

2828
from swift import get_logger
2929
from swift.utils import get_dist_setting, safe_ddp_context, subprocess_run, use_torchacc
30+
from swift.utils.torch_utils import _find_local_mac
3031
from .template import TemplateType
3132
from .utils import get_max_model_len, is_unsloth_available
3233

@@ -35,6 +36,15 @@
3536
# Model Home: 'https://modelscope.cn/models/{model_id_or_path}'
3637
MODEL_MAPPING: Dict[str, Dict[str, Any]] = {}
3738

39+
if int(os.getenv('NNODES', 1)) > 1:
40+
mac = _find_local_mac().replace(':', '')
41+
transformers.utils.hub.TRANSFORMERS_DYNAMIC_MODULE_NAME = (
42+
transformers.utils.hub.TRANSFORMERS_DYNAMIC_MODULE_NAME + mac)
43+
transformers.utils.TRANSFORMERS_DYNAMIC_MODULE_NAME = transformers.utils.hub.TRANSFORMERS_DYNAMIC_MODULE_NAME
44+
transformers.TRANSFORMERS_DYNAMIC_MODULE_NAME = transformers.utils.hub.TRANSFORMERS_DYNAMIC_MODULE_NAME
45+
transformers.dynamic_module_utils.TRANSFORMERS_DYNAMIC_MODULE_NAME = (
46+
transformers.utils.hub.TRANSFORMERS_DYNAMIC_MODULE_NAME)
47+
3848

3949
class ModelType:
4050
# qwen

0 commit comments

Comments
 (0)