From 443bc09c11f3cf89ffc573aab2021f0c933aa5b3 Mon Sep 17 00:00:00 2001
From: kmn1024 <kienman@gmail.com>
Date: 星期三, 25 六月 2025 16:34:30 +0800
Subject: [PATCH] Bugfix: Only allow rank==0 to clean up old checkpoints (#2558)

---
 funasr/frontends/default.py |    2 +-
 1 files changed, 1 insertions(+), 1 deletions(-)

diff --git a/funasr/frontends/default.py b/funasr/frontends/default.py
index 462da86..6ea88d6 100644
--- a/funasr/frontends/default.py
+++ b/funasr/frontends/default.py
@@ -20,6 +20,7 @@
 
 
 @tables.register("frontend_classes", "DefaultFrontend")
+@tables.register("frontend_classes", "EspnetFrontend")
 class DefaultFrontend(nn.Module):
     """Conventional frontend structure for ASR.
     Stft -> WPE -> MVDR-Beamformer -> Power-spec -> Mel-Fbank -> CMVN
@@ -235,7 +236,6 @@
         self, input: torch.Tensor, input_lengths: torch.Tensor
     ) -> Tuple[torch.Tensor, torch.Tensor]:
         # 1. Domain-conversion: e.g. Stft: time -> time-freq
-        # import pdb;pdb.set_trace()
         if self.stft is not None:
             input_stft, feats_lens = self._compute_stft(input, input_lengths)
         else:

--
Gitblit v1.9.1