From 837c1585738792873a6eb1583f5f33db0f7d125f Mon Sep 17 00:00:00 2001
From: ZhikangNiu <nzk020109@163.com>
Date: Sat, 23 Nov 2024 00:03:17 +0800
Subject: [PATCH] update bigvgan Dockerfile

---
 Dockerfile | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/Dockerfile b/Dockerfile
index ff5c31a..0f94cec 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -10,14 +10,17 @@ RUN set -x \
     && apt-get update \
     && apt-get -y install wget curl man git less openssl libssl-dev unzip unar build-essential aria2 tmux vim \
     && apt-get install -y openssh-server sox libsox-fmt-all libsox-fmt-mp3 libsndfile1-dev ffmpeg \
+    && apt-get install librdmacm1 libibumad3 librdmacm-dev libibverbs1 libibverbs-dev ibverbs-utils ibverbs-providers \
     && rm -rf /var/lib/apt/lists/* \
     && apt-get clean
-
+    
 WORKDIR /workspace
 
 RUN git clone https://github.com/SWivid/F5-TTS.git \
     && cd F5-TTS \
-    && pip install -e .[eval]
+    && git submodule update --init --recursive \
+    && sed -i '7iimport sys\nsys.path.append(os.path.dirname(os.path.abspath(__file__)))' src/third_party/BigVGAN/bigvgan.py \
+    && pip install -e . --no-cache-dir
 
 ENV SHELL=/bin/bash