一、官方教学
offline mode的官方教学:https://huggingface.co/docs/transformers/installation#offline-mode
二、大概步骤
- 从hugging face 里下载模型;
- 把下载好的模型文件放到指定的目录下;
- 修改环境变量,添加HF_DATASETS_OFFLINE=1;TRANSFORMERS_OFFLINE=1 或者 在.from_pretrained中添加参数 local_files_only=True。
三、具体实例
以dpt-depth-estimation-3d-obj项目为实例,具体完成一遍流程:
-
在hugging face官网中选择Models,搜索指定的模型名称:这里是Intel/dpt-large
hugging face官网:https://huggingface.co/models
-
点击Files and versions,不知道需要哪个文件,就无脑操作把所有文件全下载
-
把下载完成的文件放在指定的文件夹里,修改模型的加载路径
-
.from_pretrained()函数中加入参数local_files_only=True或者在环境变量中加入HF_DATASETS_OFFLINE=1;TRANSFORMERS_OFFLINE=1 (注意:变量之间用“;”分开,操作如下图)
加入参数
打开环境变量的方法
添加环境变量