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.

Files changed (1) hide show
  1. apps/audio_cloning/vallex/main.py +1 -1
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(OUTPUT_BASE_DIR, "models/whisper")
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