From 18c1b2f0543b42f589c708f1b96c2a2f5a8db5d4 Mon Sep 17 00:00:00 2001
From: Yifan Yang <64255737+yfyeung@users.noreply.github.com>
Date: Wed, 30 Oct 2024 19:35:56 +0800
Subject: [PATCH] Update prepare.sh

---
 egs/librilight/SSL/prepare.sh | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/egs/librilight/SSL/prepare.sh b/egs/librilight/SSL/prepare.sh
index 160c66c7da..fd1037eea6 100755
--- a/egs/librilight/SSL/prepare.sh
+++ b/egs/librilight/SSL/prepare.sh
@@ -86,15 +86,15 @@ if [ $stage -le 4 ] && [ $stop_stage -ge 4 ]; then
     wget https://dl.fbaipublicfiles.com/hubert/hubert_base_ls960_L9_km500.bin -P download
   fi
   if [ ! -e data/kmeans/.extract_small.done ]; then
-    ./local/extract_kmeans.py --subset small
+    ./local/discretize_with_s3tokenizer.py --subset small
     touch data/kmeans/.extract_small.done
   fi
   if [ ! -e data/kmeans/.extract_medium.done ]; then
-    ./local/extract_kmeans.py --subset medium
+    ./local/discretize_with_s3tokenizer.py --subset medium
     touch data/kmeans/.extract_medium.done
   fi
   if [ ! -e data/kmeans/.extract_large.done ]; then
-    ./local/extract_kmeans.py --subset large
+    ./local/discretize_with_s3tokenizer.py --subset large
     touch data/kmeans/.extract_large.done
   fi
 fi