From 01df8f330ccc754223d5e2d688dc0a55d27f2dcc Mon Sep 17 00:00:00 2001
From: querryton <72929808+querryton@users.noreply.github.com>
Date: 星期六, 20 四月 2024 16:07:13 +0800
Subject: [PATCH] [fix] Fix a bug in seaco_paraformer model "inference" function (#1639)
---
examples/industrial_data_pretraining/qwen_audio/demo_chat_from_local.py | 5 +++--
1 files changed, 3 insertions(+), 2 deletions(-)
diff --git a/examples/industrial_data_pretraining/qwen_audio/demo_chat_from_local.py b/examples/industrial_data_pretraining/qwen_audio/demo_chat_from_local.py
index d48d909..2333217 100644
--- a/examples/industrial_data_pretraining/qwen_audio/demo_chat_from_local.py
+++ b/examples/industrial_data_pretraining/qwen_audio/demo_chat_from_local.py
@@ -7,7 +7,7 @@
from funasr import AutoModel
-model = AutoModel(model="Qwen/Qwen-Audio-Chat",
+model = AutoModel(model="Qwen-Audio-Chat",
model_path="/nfs/zhifu.gzf/init_model/qwen/Qwen-Audio-Chat",
)
@@ -19,8 +19,9 @@
res = model.generate(input=audio_in, prompt=prompt, cache=cache)
print(res)
-prompt = 'Find the start time and end time of the word "middle classes"'
+
# 2nd dialogue turn
+prompt = 'Find the start time and end time of the word "middle classes"'
res = model.generate(input=None, prompt=prompt, cache=cache)
print(res)
--
Gitblit v1.9.1