Create duplicate preprocessor for data input pipeline

This commit is contained in:
mbsantiago 2026-03-19 00:03:29 +00:00
parent 91806aa01e
commit 0d90cb5cc3

View File

@ -73,9 +73,8 @@ def run_train(
audio_loader = audio_loader or build_audio_loader(config=audio_config) audio_loader = audio_loader or build_audio_loader(config=audio_config)
if model is not None: # NOTE: Create a new preprocessor instead of using the one from the model
preprocessor = preprocessor or model.preprocessor # to avoid issues with device placement.
preprocessor = preprocessor or build_preprocessor( preprocessor = preprocessor or build_preprocessor(
input_samplerate=audio_loader.samplerate, input_samplerate=audio_loader.samplerate,
config=model_config.preprocess, config=model_config.preprocess,