From cbe4753e1aadc1f48a992670435d908f85ca0397 Mon Sep 17 00:00:00 2001
From: Lance Martin <122662504+rlancemartin@users.noreply.github.com>
Date: Fri, 1 Dec 2023 15:13:20 -0800
Subject: [PATCH] Update Open CLIP embd (#14155)

Prior default model required a large amt of RAM and often crashed
Jupyter ntbk kernel.
---
 .../langchain_experimental/open_clip/open_clip.py             | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/libs/experimental/langchain_experimental/open_clip/open_clip.py b/libs/experimental/langchain_experimental/open_clip/open_clip.py
index 838148cb31..b19567d179 100644
--- a/libs/experimental/langchain_experimental/open_clip/open_clip.py
+++ b/libs/experimental/langchain_experimental/open_clip/open_clip.py
@@ -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
             )