当前位置: 首页 > article >正文

Intern大模型训练营(四):使用Hugging Face下载模型

1. Huggingface下载模型

首先在Huggingface平台注册账号。

然后进入https://github.com/codespaces,选择使用jupyter_notebook配置,输入以下命令流安装香瓜依赖。

# 安装transformers
pip install transformers==4.38
pip install sentencepiece==0.1.99
pip install einops==0.8.0
pip install protobuf==5.27.2
pip install accelerate==0.33.0

然后新建一个hf_download_josn.py 文件,输入以下内容。

import os
from huggingface_hub import hf_hub_download

# 指定模型标识符
repo_id = "internlm/internlm2_5-7b"

# 指定要下载的文件列表
files_to_download = [
    {"filename": "config.json"},
    {"filename": "model.safetensors.index.json"}
]

# 创建一个目录来存放下载的文件
local_dir = f"{repo_id.split('/')[1]}"
os.makedirs(local_dir, exist_ok=True)

# 遍历文件列表并下载每个文件
for file_info in files_to_download:
    file_path = hf_hub_download(
        repo_id=repo_id,
        filename=file_info["filename"],
        local_dir=local_dir
    )
    print(f"{file_info['filename']} file downloaded to: {file_path}")

部分代码解释如下:

  1. local_dir = f"{repo_id.split('/')[1]}": 这行代码创建了一个新变量 local_dir,它通过分割 repo_id 字符串并取其第二部分来形成模型的本地目录名称。split('/') 会根据斜杠 / 分割字符串,返回一个列表,取列表的第二个元素(索引为1)作为目录名。

  2. os.makedirs(local_dir, exist_ok=True): 这行代码使用 os 模块的 makedirs 函数创建一个新目录。local_dir 是要创建的目录的名称。exist_ok=True 参数意味着如果目录已经存在,不会抛出错误,而是正常执行。

 

运行该文件,可以看到,已经从Hugging Face上下载了相应配置文件。

 2. Hugging Face Spaces的使用

进入https://huggingface.co/spaces,在右上角点击Create new Space进行创建,在创建页面中,输入项目名为intern_cobuild,并选择Static应用进行创建。

回到codespaces,在终端输入:

git clone https://huggingface.co/spaces/<your_username>/intern_cobuild

然后再对应的文件夹下找到index文件,输入以下代码:

<!doctype html>
<html>
<head>
  <meta charset="utf-8" />
  <meta name="viewport" content="width=device-width" />
  <title>My static Space</title>
  <style>
    html, body {
      margin: 0;
      padding: 0;
      height: 100%;
    }
    body {
      display: flex;
      justify-content: center;
      align-items: center;
    }
    iframe {
      width: 430px;
      height: 932px;
      border: none;
    }
  </style>
</head>
<body>
  <iframe src="https://colearn.intern-ai.org.cn/cobuild" title="description"></iframe>
</body>
</html>

然后远程push我们更新的代码,输入以下指令:

git add .
git commit -m "update: colearn page"
git push

其中git push 会报错。

这里需要在Huggingface上获取一个Access token,在Huggingface用户->settings->Access Tokens新建一个token。

然后输入以下

git remote set-url origin https://<user_name>:<token>@huggingface.co/<repo_path>

之后再次git push,是成功的。

回到Huggingface的spaces页面,可以看到页面已经被更新。

3. 上传模型

通过CLI上传 Hugging Face同样是跟Git相关联,通常大模型的模型文件都比较大,因此我们需要安装git lfs,对大文件系统支持。

curl -s https://packagecloud.io/install/repositories/github/git-lfs/script.deb.sh | sudo bash
git lfs install # 直接在git环境下配置git LFS
pip install huggingface_hub

创建项目,注意这里不要在之前的intern_build仓库里面创建,本人在这里踩了坑,没注意父文件夹。

cd /workspaces/codespaces-jupyter

#intern_study_L0_4就是model_name
huggingface-cli repo create intern_study_L0_4

# 克隆到本地 your_github_name 注意替换成你自己的
git clone https://huggingface.co/{your_github_name}/intern_study_L0_4

在 intern_study_L0_4 上传之前下载好的config.json,然后新建一个README.md,粘贴以下内容。

# 书生浦语大模型实战营camp4
- hugging face模型上传测试
- 更多内容请访问 https://github.com/InternLM/Tutorial/tree/camp4

现在可以用git提交到远程仓库

cd intern_study_L0_4
git add .
git commit -m "add:intern_study_L0_4"
git push

现在可以在Hugging Face的个人profile里面看到这个model。


http://www.kler.cn/a/389329.html

相关文章:

  • MySQL技巧之跨服务器数据查询:进阶篇-从A数据库复制到B数据库的表中
  • 《MYSQL45讲》误删数据怎么办
  • MySQL45讲 第二十讲 幻读是什么,幻读有什么问题?
  • MTSET可溶于DMSO、DMF、THF等有机溶剂,并在水中有轻微的溶解性,91774-25-3
  • Toeplitz矩阵循环矩阵
  • 深入理解接口测试:实用指南与最佳实践5.0(二)
  • RoseTTAFold PositionalEncoding类解读
  • (C++11)委托构造函数--C++
  • 如何在Oracle应用中使用BI PUBLISHER API将RTF转换为XSL-FO
  • XGBoost算法Python代码实现
  • Iceberg 写入和更新模式,COW,MOR(Copy-on-Write,Merge-on-Read)
  • sqli-labs(第二关)
  • ThinkBook 14+ 2024 Ubuntu 触控板失效 驱动缺失问题解决
  • 腾讯自研的 Git 客户端!!【送源码】
  • vue3中使用输入框按回车键刷新页面问题
  • 零拷贝Zero Copy
  • RabbitMQ的应用
  • 【STK学习】part1-卫星轨道与Walker星座基础知识
  • 《TCP/IP网络编程》学习笔记 | Chapter 4:基于TCP的服务器端/客户端(1)
  • 信息泄露漏洞一文速通
  • Facebook vs. Google:哪个更适合你的品牌
  • rabbitMq怎么保证消息不丢失?消费者没有接收到消息怎么处理
  • MySQL_聚合函数分组查询
  • 【神经科学学习笔记】基于分层嵌套谱分割(Nested Spectral Partition)模型分析大脑网络整合与分离的学习总结
  • 无人机手势控制工作原理和算法!
  • 【笔记】开关电源变压器设计 - 工作磁通的选择原则