下载huggingface-transformers模型

最新推荐文章于 2025-03-30 23:07:09 发布

x_cube

最新推荐文章于 2025-03-30 23:07:09 发布

阅读量3.6k

点赞数 5

文章标签：数学建模

本文链接：https://blog.csdn.net/qq_41458274/article/details/131134683

版权

为何不调用from_pretrained方法直接下载模型？而且只要联网就可用。但是

如果网络很不好，模型下载时间会很久，一个小模型下载几个小时也很常见
如果换了训练服务器，又要重新下载。
而且容易报错

如何调用huagging face 里的模型，例如vit

本地可以在Models - Hugging Face找到你需要的模型下载在本地进行调用。例子如下

from transformers import ViTImageProcessor, ViTModel
from PIL import Image
import requests

url = 'http://images.cocodataset.org/val2017/000000039769.jpg'
image = Image.open(requests.get(url, stream=True).raw)
# 此处可以是模型名字也可以选择本地路径，本地不容易报错
image_model = 'google/vit-base-patch16-224-in21k'
# image_model = ./models/vit-base-patch16-224-in21k'

processor = ViTImageProcessor.from_pretrained(image_model)
model = ViTModel.from_pretrained(image_model)
inputs = processor(images=image, return_tensors="pt")

outputs = model(**inputs)
last_hidden_states = outputs.last_hidden_state

下载操作：

1打开模型页面，输入你要的模型。