19 lines
458 B
Python
19 lines
458 B
Python
|
|
print('hello')
|
|
# %%
|
|
import open_clip
|
|
import torch
|
|
|
|
do_load = True
|
|
if do_load:
|
|
|
|
|
|
model_name = 'hf-hub:timm/ViT-L-16-SigLIP2-512'
|
|
# pretrained_name = 'webli'
|
|
model, preprocess = open_clip.create_model_from_pretrained('hf-hub:timm/ViT-L-16-SigLIP2-512')
|
|
|
|
# model, _, preprocess = open_clip.create_model_and_transforms(model_name, pretrained=pretrained_name)
|
|
device = 'cpu'
|
|
model.eval()
|
|
tokenizer = open_clip.get_tokenizer(model_name)
|