Update Open CLIP embd (#14155)

Prior default model required a large amt of RAM and often crashed
Jupyter ntbk kernel.
pull/14163/head
Lance Martin 7 months ago committed by GitHub
parent b01d9d27d9
commit cbe4753e1a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -19,8 +19,8 @@ class OpenCLIPEmbeddings(BaseModel, Embeddings):
# model_name = "ViT-B-32"
# checkpoint = "laion2b_s34b_b79k"
### Larger, more performant
model_name = "ViT-g-14"
checkpoint = "laion2b_s34b_b88k"
model_name = "ViT-H-14"
checkpoint = "laion2b_s32b_b79k"
model, _, preprocess = open_clip.create_model_and_transforms(
model_name=model_name, pretrained=checkpoint
)

Loading…
Cancel
Save