当前位置:   article > 正文

huggingface.co下载模型_blip模型下载

blip模型下载

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

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

  1. import pdb /from PIL import Imagefrom clip_interrogator import Interrogator, Config
  2. image_path = "shot142.jpg"
  3. image = Image.open(image_path).convert('RGB')
  4. 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行有限判断下载,再读是否存在本地文件(求大神指教这是什么神仙逻辑?)

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

5、成功加载

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/寸_铁/article/detail/737396
推荐阅读
相关标签
  

闽ICP备14008679号