From 399208b7bb83ebdd11fc45e7b4be85fba5ebf0a9 Mon Sep 17 00:00:00 2001
From: yihanzhao <yihan@marqo.ai>
Date: Tue, 27 Aug 2024 09:53:50 +1000
Subject: [PATCH] Add dependencies for LanguageBind model

---
 requirements.txt             | 8 ++++++++
 scripts/install_torch_amd.sh | 2 +-
 2 files changed, 9 insertions(+), 1 deletion(-)

diff --git a/requirements.txt b/requirements.txt
index f73e946..99dd8e9 100644
--- 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
\ No newline at end of file
diff --git a/scripts/install_torch_amd.sh b/scripts/install_torch_amd.sh
index 33f9ec3..7696203 100644
--- 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
\ No newline at end of file