liumaolin commited on
Commit
4c23586
·
1 Parent(s): c5cc64b

Disable FunASR pbar in Warmup.

Browse files
Files changed (1) hide show
  1. transcribe/helpers/funasr.py +1 -1
transcribe/helpers/funasr.py CHANGED
@@ -24,7 +24,7 @@ class FunASR:
24
  def warmup(self, warmup_steps=1):
25
  warmup_soundfile = f"{config.ASSERT_DIR}/jfk.flac"
26
  for _ in range(warmup_steps):
27
- self.model.generate(input=warmup_soundfile)
28
 
29
  def transcribe(self, audio_buffer: bytes, language):
30
  audio_frames = np.frombuffer(audio_buffer, dtype=np.float32)
 
24
  def warmup(self, warmup_steps=1):
25
  warmup_soundfile = f"{config.ASSERT_DIR}/jfk.flac"
26
  for _ in range(warmup_steps):
27
+ self.model.generate(input=warmup_soundfile, disable_pbar=True)
28
 
29
  def transcribe(self, audio_buffer: bytes, language):
30
  audio_frames = np.frombuffer(audio_buffer, dtype=np.float32)