From 5f40262a96f1fac361a8f8869f6fb38ff7647ec1 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E5=BC=A0=E6=98=A5=E4=B9=94?= <83450930+Liyulingyue@users.noreply.github.com> Date: Fri, 8 Nov 2024 14:32:27 +0800 Subject: [PATCH] Fix wav2vec error in Demos/ssl (#3872) * Update infer.py * Update wav2vec2_ASR.py * Apply suggestions from code review * Apply suggestions from code review * Update infer.py * Update wav2vec2_ASR.py * Apply suggestions from code review --- paddlespeech/cli/ssl/infer.py | 18 ++++++++++++++++++ 1 file changed, 18 insertions(+) diff --git a/paddlespeech/cli/ssl/infer.py b/paddlespeech/cli/ssl/infer.py index 9b4b02803..33cdf7637 100644 --- a/paddlespeech/cli/ssl/infer.py +++ b/paddlespeech/cli/ssl/infer.py @@ -119,6 +119,7 @@ class SSLExecutor(BaseExecutor): '--verbose', action='store_true', help='Increase logger verbosity of current task.') + self.last_call_params = None def _init_from_path(self, model_type: str=None, @@ -453,6 +454,23 @@ class SSLExecutor(BaseExecutor): Python API to call an executor. """ + current_call_params = { + "model": model, + "task": task, + "lang": lang, + "sample_rate": sample_rate, + "config": config, + "ckpt_path": ckpt_path, + "decode_method": decode_method, + "force_yes": force_yes, + "rtf": rtf, + "device": device + } + if self.last_call_params is not None and self.last_call_params != current_call_params and hasattr( + self, 'model'): + del self.model + self.last_call_params = current_call_params + audio_file = os.path.abspath(audio_file) paddle.set_device(device) self._init_from_path(model, task, lang, sample_rate, config,