github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-06-29 01:48:57 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update Open CLIP embd (#14155)

Browse Source 
Prior default model required a large amt of RAM and often crashed
Jupyter ntbk kernel.
This commit is contained in:
Lance Martin 2023-12-01 15:13:20 -08:00 committed by GitHub
parent b01d9d27d9
commit cbe4753e1a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
libs/experimental/langchain_experimental/open_clip/open_clip.py
View File

@ -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
)