Spaces:
Sleeping
Sleeping
nukopy commited on
Commit ·
71c3f53
1
Parent(s): 0a9541c
fix: update output directory for Whisper model
Browse files- Changed the output directory for the Whisper model from OUTPUT_BASE_DIR to PREPARED_BASE_DIR to ensure proper model storage.
apps/audio_cloning/vallex/main.py
CHANGED
|
@@ -163,7 +163,7 @@ audio_tokenizer = AudioTokenizer(device)
|
|
| 163 |
vocos = Vocos.from_pretrained("charactr/vocos-encodec-24khz").to(device)
|
| 164 |
|
| 165 |
# initialize ASR model
|
| 166 |
-
OUTPUT_DIR_WHISPER = os.path.join(
|
| 167 |
if not os.path.exists(OUTPUT_DIR_WHISPER):
|
| 168 |
os.makedirs(OUTPUT_DIR_WHISPER, exist_ok=True)
|
| 169 |
|
|
|
|
| 163 |
vocos = Vocos.from_pretrained("charactr/vocos-encodec-24khz").to(device)
|
| 164 |
|
| 165 |
# initialize ASR model
|
| 166 |
+
OUTPUT_DIR_WHISPER = os.path.join(PREPARED_BASE_DIR, "models/whisper")
|
| 167 |
if not os.path.exists(OUTPUT_DIR_WHISPER):
|
| 168 |
os.makedirs(OUTPUT_DIR_WHISPER, exist_ok=True)
|
| 169 |
|