clip加载text encoder
【代码】clip加载text encoder。
·
import os
import clip
import torch
# Load the model
device = "cuda" if torch.cuda.is_available() else "cpu"
model, preprocess = clip.load('ViT-B/32', device)
text_inputs = clip.tokenize('hello').to(device)
with torch.no_grad():
text_features = model.encode_text(text_inputs)
text_features /= text_features.norm(dim=-1, keepdim=True)
更多推荐
所有评论(0)