Added torchvision dependency

TJ-Solergibert · TJ-Solergibert · commit 9a02575a6f0c · 2024-11-06T13:41:03.000+01:00
diff --git a/.ci/docker/requirements.txt b/.ci/docker/requirements.txt
@@ -6,3 +6,4 @@ sentencepiece
 tiktoken
 blobfile
 tabulate
+torchvision
diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml
diff --git a/pyproject.toml b/pyproject.toml
@@ -18,6 +18,9 @@ dependencies = [
     "sentencepiece",
     "tiktoken",
 
+    # Multimodality
+    "torchvision",
+
     # Miscellaneous
     "tomli>=1.1.0"
 ]
