china_download_hugingface

# huggingface模型下载到国内本地NAS共享使用的方法

linux下载到nas 共享模型的方法

• linux 下安装smb协议

  sudo apt update
  sudo apt install samba
  sudo systemctl enable smbd
  

• 设置映射 修改fstab 加入如下内容 具体ip及路径 参考自己的nas 设置 //192.168.1.123/public/ /mnt/public cifs credentials=/etc/samba/credentials,file_mode=0777,dir_mode=0777 0 0

• 编辑/etc/samba/credentials 文件 文件内容如下

  user=xxxxx   
  password=xxxxx 
  

• 安装python3.6 及以上

• 安装 huggingface_cli

  pip install huggingface-hub
  

• 编写 linux 下 shell dw_model.sh

#!/bin/bash    
#设置环境变量
export HF_HOME=/mnt/public/huggingface
export HF_ENDPOINT=https://hf-mirror.com                                                                        
#模型名称作为参数                                                                     
MODEL_NAME=$1
#检查是否提供了模型名称                                                               
if [ -z "$MODEL_NAME" ]; then                                                          
    echo "错误：没有提供模型名称。"
    echo "用法：$0 [模型名称]"
    exit 1
fi                                                                                     
#使用 huggingface-cli 下载模型                                                        
huggingface-cli download --resume-download --local-dir-use-symlinks False $MODEL_NAME    

保存shell 为 dw_model.sh

执行 .\dw_model.sh xxxxxx\xxxxx 就可以了

windows下载到nas 共享模型的方法

• 安装python3.6 及以上

• 安装 huggingface_cli

  pip install huggingface-hub
  

• 编辑dw_model.bat 批处理文件

@echo off
rem 设置环境变量
set HF_HOME=\\192.168.1.9\public\huggingface
set HF_ENDPOINT=https://hf-mirror.com

rem 检查是否提供了模型名称
if "%~1"=="" (
    echo 错误: 没有提供模型名称。
    echo 用法: %0 [模型名称]
    exit /b 1
)

rem 使用 huggingface-cli 下载模型
huggingface-cli download --resume-download --local-dir-use-symlinks False  %1