Add dependencies for LanguageBind model

marqo-ai · Aug 26, 2024 · 399208b · 399208b
1 parent 81346ad
commit 399208b
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/requirements.txt b/requirements.txt
@@ -14,6 +14,7 @@ botocore==1.28.4
 nltk==3.7
 torch==1.12.1
 torchvision==0.13.1
+torchaudio==0.12.1
 Pillow==9.3.0
 numpy==1.23.4
 validators==0.20.0
@@ -38,3 +39,10 @@ urllib3==1.26.16
 timm==1.0.8
 transformers==4.41.2
 safetensors==0.4.1
+decord==0.6.0
+einops==0.6.1
+soundfile==0.12.1
+python-magic==0.4.27
+ffmpeg-python==0.2.0
+librosa==0.10.2.post1
+pytorchvideo==0.1.5
diff --git a/scripts/install_torch_amd.sh b/scripts/install_torch_amd.sh
@@ -7,5 +7,5 @@ if ! [[ "$TARGETPLATFORM" ]]; then
 fi
 
 if [[ "$TARGETPLATFORM" != "linux/arm64" ]]; then
-  pip3 install --no-cache-dir torch==1.12.1+cu113 torchvision==0.13.1+cu113 torchaudio==0.12.1 --extra-index-url https://download.pytorch.org/whl/cu113 --upgrade
+  pip3 install --no-cache-dir torch==1.12.1+cu113 torchvision==0.13.1+cu113 --extra-index-url https://download.pytorch.org/whl/cu113 --upgrade
 fi