huggingface.co下载模型

popoala

已于 2022-12-02 13:59:38 修改

阅读量4.9k

点赞数

分类专栏： transformer 文章标签： python java 开发语言

于 2022-12-02 13:56:53 首次发布

本文链接：https://blog.csdn.net/popoala/article/details/128146763

版权

transformer 专栏收录该内容

1 篇文章 0 订阅

订阅专栏

1、在测试clip-interrogator 发现BLIP CLIP模型需要离线加载，以下手工下载并指定路径。

并运行TRANSFORMERS_OFFLINE=1 python run.py, run.py如下

import pdb                                                                                                 /from PIL import Imagefrom clip_interrogator import Interrogator, Config
image_path = "shot142.jpg"
image = Image.open(image_path).convert('RGB')
ci = Interrogator(Config(clip_model_name="ViT-L-14/openai"))print(ci.interrogate(image))

其他huggingface的模型处理方法可参考。

2、BLIP模型根据Readme指引下载，存，指定路径即可。

BLIP//pretrained/model_base_caption_capfilt_large.pth BLIP//pretrained/model_large_caption.pth

3、缺bert-base-uncased 模型，尝试一下方法下载

——git clone https://huggingface.co/bert-base-uncased

然后在文中'bert-base-uncased'位置指定为下载路径

4、代码中只有 clip_model_name: str = 'ViT-L-14/openai' 一行提示模型，无法通过上述clone下载，

——从出错位置跟踪找到模型下载路径，手工下载并存到指定路径

并且！！对环境中py379/lib/python3.7/site-packages/open_clip/openai.py的相关调用如58行有限判断下载，再读是否存在本地文件（求大神指教这是什么神仙逻辑？）

if os.path.isfile(name):
        print(name)
        model_path = name
    elif get_pretrained_url(name, 'openai'):
        print(name, get_pretrained_url(name, 'openai'), cache_dir)
        model_path = download_pretrained_from_url(get_pretrained_url(name, 'openai'), cache_dir=cache_dir)
    else:
        raise RuntimeError(f"Model {name} not found; available models = {list_openai_models()}")

5、成功加载