Update OpenCLIP docs (#14319)

This commit is contained in:
Lance Martin
2023-12-05 15:31:10 -08:00
committed by GitHub
parent a74c03da3c
commit 29e993a5f2

View File

@@ -92,7 +92,7 @@
"uri_house = \"/Users/rlm/Desktop/test/house.jpg\"\n",
"\n",
"# Embe images or text\n",
"clip_embd = OpenCLIPEmbeddings()\n",
"clip_embd = OpenCLIPEmbeddings(model_name=\"ViT-g-14\", checkpoint=\"laion2b_s34b_b88k\")\n",
"img_feat_dog = clip_embd.embed_image([uri_dog])\n",
"img_feat_house = clip_embd.embed_image([uri_house])\n",
"text_feat_dog = clip_embd.embed_documents([\"dog\"])\n",