文件名称:CLIP-pytorch:pytorch中OpenAI的CLIP的非JIT版本实现复制
文件大小:230KB
文件格式:ZIP
更新时间:2024-06-16 22:01:11
pytorch clip image-text-matching Python
CLIP模型文件和预训练的state_dict 更新01/16/2021 ::::目前,官方代码库发布了非JIT模型。 您可以参考标准实现。 的复制。 由于官方存储库仅提供JIT版本模型(VIT-B-32) ,因此修改不方便。 这是一个非JIT版本模型。 这样,您可以随时进行更改。 用法 安装 请参考官方仓库 ,并检查。 下面的$ {BPEPATH}在,$ {MODELPATH}是从JIT版本模型中提取的state_dict,并提供 推理 在此处使用它进行推理的示例: from CLIP import CLIP , build_transform from clip_tokenizer import SimpleTokenizer from PIL import Image import torch model = CLIP ( attention_probs_dropout_p
【文件预览】:
CLIP-pytorch-main
----CLIP.png(247KB)
----clip_tokenizer.py(5KB)
----CLIP.py(13KB)
----readme.md(2KB)