Add ViT-L/14.

nshepperd · Jan 26, 2022 · a858a7e · a858a7e
1 parent 7abbd61
commit a858a7e
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/clip_jax/clip.py b/clip_jax/clip.py
@@ -25,6 +25,7 @@
     # "RN50": "https://openaipublic.azureedge.net/clip/models/afeb0e10f9e5a86da6080e35cf09123aca3b358a0c3e3b6c78a7b63bc04b6762/RN50.pt",
     "ViT-B/32": "https://openaipublic.azureedge.net/clip/models/40d365715913c9da98579312b702a82c18be219cc2a73407c4526f58eba950af/ViT-B-32.pt",
     "ViT-B/16": "https://openaipublic.azureedge.net/clip/models/5806e77cd80f8b59890b7e101eabd078d9fb84e6937f9e85e4ecb61988df416f/ViT-B-16.pt",
+    "ViT-L/14": "https://openaipublic.azureedge.net/clip/models/b8cca3fd41ae0c99ba7e8951adf17d267cdb84cd88be6f7c2e0eca1737a03836/ViT-L-14.pt",
 }
 
 

diff --git a/tests/test_consistency.py b/tests/test_consistency.py
@@ -27,4 +27,5 @@ def test_model(model_name):
     print(f"{model_name}: done!")
 
 test_model("ViT-B/32")
-test_model("ViT-B/16")
+test_model("ViT-B/16")
+test_model("ViT-L/14")