chore: sample code runs

xhiroga · Oct 25, 2024 · df0eef2 · df0eef2
1 parent 596e080
commit df0eef2
Show file tree

Hide file tree

Showing 4 changed files with 57 additions and 49 deletions.
diff --git a/software-engineering/openai/clip/_src/text-encoder-only/README.md b/software-engineering/openai/clip/_src/text-encoder-only/README.md
@@ -0,0 +1,10 @@
+# Use Text Encoder Only
+
+```shell
+uv sync
+uv run python main.py
+```
+
+## References
+
+- <https://github.com/openai/CLIP/issues/113>
diff --git a/software-engineering/openai/clip/_src/text-encoder-only/main.py b/software-engineering/openai/clip/_src/text-encoder-only/main.py
@@ -0,0 +1,11 @@
+import torch
+import clip
+
+device = "cuda" if torch.cuda.is_available() else "cpu"
+model, preprocess = clip.load("ViT-B/32", device=device)
+
+text = clip.tokenize(["a diagram", "a dog", "a cat"]).to(device)
+
+with torch.no_grad():
+    text_features = model.encode_text(text)
+    print(f"{text_features.shape}")
diff --git a/software-engineering/openai/clip/_src/text-encoder-only/pyproject.toml b/software-engineering/openai/clip/_src/text-encoder-only/pyproject.toml
@@ -4,12 +4,13 @@ version = "0.1.0"
 description = "Add your description here"
 readme = "README.md"
 requires-python = ">=3.12"
+# torch==2.5.0 causes "Error: No execution plans support the graph."
 dependencies = [
     "clip",
     "ftfy>=6.3.0",
     "regex>=2024.9.11",
-    "torch==2.5.0+cu124",
-    "torchvision==0.20.0+cu124",
+    "torch==2.4.1+cu124",
+    "torchvision==0.19.1+cu124",
     "tqdm>=4.66.5",
 ]
 

diff --git a/software-engineering/openai/clip/_src/text-encoder-only/uv.lock b/software-engineering/openai/clip/_src/text-encoder-only/uv.lock