From f7fc2f32a4e8d2b0f7c846bd227188c1df261616 Mon Sep 17 00:00:00 2001 From: Kuray107 Date: Thu, 15 Aug 2024 19:13:41 +0000 Subject: [PATCH] Apply isort and black reformatting Signed-off-by: Kuray107 --- examples/audio/audio_to_audio_train.py | 2 +- nemo/collections/audio/models/enhancement.py | 1 + 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/examples/audio/audio_to_audio_train.py b/examples/audio/audio_to_audio_train.py index b15b43dcbcbf3..cef46dcf20b6d 100644 --- a/examples/audio/audio_to_audio_train.py +++ b/examples/audio/audio_to_audio_train.py @@ -51,7 +51,7 @@ class ModelType(str, Enum): Predictive = 'predictive' ScoreBased = 'score_based' SchroedingerBridge = 'schroedinger_bridge' - FlowMatching= 'flow_matching' + FlowMatching = 'flow_matching' def get_model_class(model_type: ModelType): diff --git a/nemo/collections/audio/models/enhancement.py b/nemo/collections/audio/models/enhancement.py index fa715cbf3ab9d..cd9f47b98096e 100644 --- a/nemo/collections/audio/models/enhancement.py +++ b/nemo/collections/audio/models/enhancement.py @@ -631,6 +631,7 @@ class FlowMatchingAudioToAudioModel(AudioToAudioModel): - decoder: transforms sampler output into the time domain (synthesis transform) - ssl_pretrain_masking: if it is defined, perform the ssl pretrain masking for self reconstruction in the training process """ + def __init__(self, cfg: DictConfig, trainer: Trainer = None): super().__init__(cfg=cfg, trainer=trainer) self.sample_rate = self._cfg.sample_rate