当前位置:   article > 正文

小白只需两步,优雅下载HuggingFace模型文件_下载huggingface的模型到本地python代码

下载huggingface的模型到本地python代码
- 使用工具:Python, 系统:windows10,11

第一步,把自定本地路径写入环境:

 - 打开windows环境变量

 - 然后,往外一层层点确认

第二步,download.py 下载代码

 - 在任何路径创建空白py文件,打开,写入:
  1. from huggingface_hub import snapshot_download
  2. snapshot_download(
  3. repo_id="Qwen/Qwen1.5-14B",
  4. # local_dir=".", # 注释掉直接下载到HF_HOME
  5. proxies={"https": "http://localhost:7890"}, # 选你的魔法端口
  6. max_workers=8
  7. )

- 还没安装huggingface_hub的,终端执行 pip install huggingface_hub

- 代码解释:

repo_id:抱脸的页面,运行结果是把Files内多有文档都下载到local_dir。

local_dir:可以填写你另外指定的本地保存路径,我的习惯是有专门一个路径放模型文件,直接在文件夹保存这个download.py,运行结果就是自动建立‘Qwen-14B-Chat-Int4’文件夹。

proxies={"https": "http://localhost:7890"}:一般魔法工具的端口号,可以按你的魔法端口号改写

max_workers:按需设置

- 每次复制粘贴huggingface模型路径

比如点旁边这个按钮,复制 'Qwen/Qwen1.5-14B', 写到代码 repo_id="Qwen/Qwen1.5-14B",

运行代码,即把路径里面所有文件一次下载到你的HF_HOME路径下。

- 如果中途网络出问题,下载被打断:

你只需要再次运行这份代码,自动断点续存,不会出现重复文件,也不会浪费太多流量。

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/神奇cpp/article/detail/943082
推荐阅读
相关标签
  

闽ICP备14008679号