File tree Expand file tree Collapse file tree 1 file changed +10
-0
lines changed Expand file tree Collapse file tree 1 file changed +10
-0
lines changed Original file line number Diff line number Diff line change 2727
2828from swift import get_logger
2929from swift .utils import get_dist_setting , safe_ddp_context , subprocess_run , use_torchacc
30+ from swift .utils .torch_utils import _find_local_mac
3031from .template import TemplateType
3132from .utils import get_max_model_len , is_unsloth_available
3233
3536# Model Home: 'https://modelscope.cn/models/{model_id_or_path}'
3637MODEL_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
3949class ModelType :
4050 # qwen
You can’t perform that action at this time.
0 commit comments