本地模型一键上传modelscope

Mr.Lee jack

已于 2024-06-15 14:59:12 修改

阅读量229

点赞数 4

分类专栏： Ai 文章标签： language model hugging face

于 2024-06-15 14:49:58 首次发布

本文链接：https://blog.csdn.net/xzpdxz/article/details/139702343

版权

Ai 专栏收录该内容

16 篇文章 0 订阅

订阅专栏

ModelScope官网主页

存放大模型大文件

安装依赖

apt install git-lfs
pip3 install modelscope

上传

from modelscope.hub.api import HubApi
import os, json

YOUR_ACCESS_TOKEN = '请从ModelScope个人中心->访问令牌获取'
MODEL_ID = "yourname/your_model_id"
LOCAL_MODEL_DIR = "my_model_dir"


configuration = os.path.join(LOCAL_MODEL_DIR, "configuration.json")
with open(configuration, "w") as f:
    f.write(json.dumps({"framework":"Pytorch","task":"other"}))


api = HubApi()
api.login(YOUR_ACCESS_TOKEN)
api.push_model(
    model_id=MODEL_ID , 
    model_dir=LOCAL_MODEL_DIR 
)