基本思想:在昇腾910C上适配chinese-roberta-wwm-ext
镜像地址:crpi-8ew3ouqcvy9yujug.cn-hangzhou.personal.cr.aliyuncs.com/sxj731533730/chinese-roberta-wwm-ext_910c:v0
一、下载模型
[root@hostname-g6ilx data]# cat down.py #模型下载 from modelscope import snapshot_download model_dir = snapshot_download('dienstag/chinese-roberta-wwm-ext',cache_dir=".") [root@hostname-g6ilx data]#设备类型
二、模型转onnx
import os import shutil import numpy as np import torch from transformers import AutoModelForMaskedLM, AutoTokenizer # ---- 路径配置 ------------------------------------------------------------- HERE = os.path.dirname(os.path.abspath(__