解决Hugging Face脚本下载后路径不是默认缓存的问题的小脚本分享
这篇文章介绍了一个小脚本,用于解决在使用hfd脚本下载文件后,文件路径不是Hugging Face默认缓存路径的问题。该脚本首先检查是否提供了路径参数,并验证该路径是否存在。接着,脚本会尝试自动获取最新的git commit ID,如果无法自动获取,则会提示用户手动输入。脚本会创建一个以父目录和当前目录命名的目标目录,并在其中创建blobs、refs和snapshots文件夹。在refs文件夹中,会创建一个main文件,并写入commit ID。在snapshots文件夹中,会创建以commit ID命名的子文件夹。脚本会遍历指定路径下的所有文件,根据文件大小选择不同的哈希算法,并将文件复制到目标目录的blobs目录中,同时创建相应的软链接。最后,脚本会将创建的目录结构移动到Hugging Face的缓存目录中,并输出最终的模型路径和模型名称。使用这个脚本后,用户不再需要显式指定模型路径,从而简化了使用Hugging Face相关代码的过程。
评论已关闭