书生L0G4000

任务1 模型下载

示例代码：

import os
from huggingface_hub import hf_hub_download,list_repo_files

# 指定模型标识符
repo_id = "internlm/internlm2_5-1_8b-chat"

# 指定要下载的文件列表
files_to_download = list_repo_files(repo_id)
# 创建一个目录来存放下载的文件
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,
        local_dir=local_dir
    )
    print(f"{file_info} file downloaded to: {file_path}")

关键：下载模型需要模型名称（repo_id），文件列表（filename），和指定目录（local_dir）三个参数，通过list_repo_files(repo_id)函数拉取所有文件后遍历下载即可。

运行截图：

成功后示例：