首页从huggingface下载模型

从huggingface下载模型

时间: 2025-01-07 18:59:31 浏览: 118

### 下载Hugging Face上的预训练模型为了从Hugging Face下载预训练模型，用户可以选择两种主要方式来获取这些资源。一种是从官方网站直接在线加载模型，另一种则是离线环境下的手动下载。对于在线环境中，可以直接通过Python代码调用`transformers`库中的类方法完成模型的自动下载与缓存： ```python from transformers import AutoModel, AutoTokenizer model_name = "bert-base-chinese" tokenizer = AutoTokenizer.from_pretrained(model_name) model = AutoModel.from_pretrained(model_name) ``` 这段代码会根据指定的名字（如`bert-base-chinese`），自动连接到Hugging Face服务器并下载对应的预训练权重文件以及配置文件[^1]。如果是在无法联网的情况下工作，则需要先访问[Hugging Face Models页面](https://huggingface.co/models)，挑选合适的模型版本进行下载。下载完成后解压至本地目录，在后续程序里就可以利用绝对路径或相对路径作为参数传递给`.from_pretrained()`函数实现加载操作[^3]。例如，假设已经把模型放置到了名为`./local_model_directory/`的文件夹下，那么可以这样加载它： ```python from transformers import T5Tokenizer, T5EncoderModel tokenizer = T5Tokenizer.from_pretrained("./local_model_directory/") model = T5EncoderModel.from_pretrained("./local_model_directory/") ```

阅读全文