From 53ec156198562039152093be4d1c9c9f57fe2116 Mon Sep 17 00:00:00 2001
From: zr_jin <zengrui.jin0@gmail.com>
Date: Mon, 4 Nov 2024 15:20:50 +0800
Subject: [PATCH] updated

---
 egs/ljspeech/TTS/matcha/monotonic_align/setup.py |  2 +-
 egs/ljspeech/TTS/prepare.sh                      | 16 +++++++++-------
 2 files changed, 10 insertions(+), 8 deletions(-)

diff --git a/egs/ljspeech/TTS/matcha/monotonic_align/setup.py b/egs/ljspeech/TTS/matcha/monotonic_align/setup.py
index 0c5c57070c..beacf2e369 100644
--- a/egs/ljspeech/TTS/matcha/monotonic_align/setup.py
+++ b/egs/ljspeech/TTS/matcha/monotonic_align/setup.py
@@ -1,4 +1,4 @@
-# Copied from
+# Modified from
 # https://github.com/shivammehta25/Matcha-TTS/blob/main/matcha/utils/monotonic_align/setup.py
 from Cython.Build import cythonize
 from setuptools import Extension, setup
diff --git a/egs/ljspeech/TTS/prepare.sh b/egs/ljspeech/TTS/prepare.sh
index db0d6857d6..ec5062933e 100755
--- a/egs/ljspeech/TTS/prepare.sh
+++ b/egs/ljspeech/TTS/prepare.sh
@@ -26,13 +26,15 @@ log "dl_dir: $dl_dir"
 
 if [ $stage -le -1 ] && [ $stop_stage -ge -1 ]; then
   log "Stage -1: build monotonic_align lib (used by vits and matcha recipes)"
-  if [ ! -d vits/monotonic_align/build ]; then
-    cd vits/monotonic_align
-    python3 setup.py build_ext --inplace
-    cd ../../
-  else
-    log "monotonic_align lib for vits already built"
-  fi
+  for recipe in vits matcha; do
+    if [ ! -d $recipe/monotonic_align/build ]; then
+      cd $recipe/monotonic_align
+      python3 setup.py build_ext --inplace
+      cd ../../
+    else
+      log "monotonic_align lib for $recipe already built"
+    fi
+  done
 fi
 
 if [ $stage -le 0 ] && [ $stop_stage -ge 0 ]; then