AI开发平台MODELARTS-迁移适配:自动迁移适配

时间:2024-09-14 22:29:36

自动迁移适配

修改“ptuning/main.py”,添加deepspeed_npu、torch_npu、transfer_to_npu依赖库,如下图所示。
# 导入deepspeed_npu和torch_npu
import deepspeed_npu
import torch_npu
# 导入一键迁移接口
from torch_npu.contrib import transfer_to_npu
图1 自动迁移适配
support.huaweicloud.com/bestpractice-modelarts/modelarts_10_4002.html