From 9b4e9cc8a0311e5243d69b73ed073e7ea441982e Mon Sep 17 00:00:00 2001
From: 游雁 <zhifu.gzf@alibaba-inc.com>
Date: 星期三, 27 三月 2024 16:05:29 +0800
Subject: [PATCH] train update
---
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