From ef1d7b3f12535d73bf9046085d14d192c54cf7da Mon Sep 17 00:00:00 2001
From: Yuekai Zhang <zhangyuekai@foxmail.com>
Date: 星期二, 12 十一月 2024 17:23:51 +0800
Subject: [PATCH] Merge pull request #2206 from yijinsheng/triton_gpu
---
examples/industrial_data_pretraining/qwen_audio/demo_chat.py | 10 ++++------
1 files changed, 4 insertions(+), 6 deletions(-)
diff --git a/examples/industrial_data_pretraining/qwen_audio/demo_chat.py b/examples/industrial_data_pretraining/qwen_audio/demo_chat.py
index 4fb5e5b..26b5b89 100644
--- a/examples/industrial_data_pretraining/qwen_audio/demo_chat.py
+++ b/examples/industrial_data_pretraining/qwen_audio/demo_chat.py
@@ -7,20 +7,18 @@
from funasr import AutoModel
-model = AutoModel(model="Qwen/Qwen-Audio-Chat",
- model_path=None,
- )
+model = AutoModel(model="Qwen/Qwen-Audio-Chat")
audio_in = "https://github.com/QwenLM/Qwen-Audio/raw/main/assets/audio/1272-128104-0000.flac"
# 1st dialogue turn
-prompt = 'what does the person say?'
+prompt = "what does the person say?"
cache = {"history": None}
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