From 12496e559feea69af2e77eac6f22b32df3bf6762 Mon Sep 17 00:00:00 2001
From: zhifu gao <zhifu.gzf@alibaba-inc.com>
Date: 星期四, 18 一月 2024 23:21:12 +0800
Subject: [PATCH] streaming bugfix (#1271)
---
examples/industrial_data_pretraining/ct_transformer_streaming/demo.py | 2 +-
1 files changed, 1 insertions(+), 1 deletions(-)
diff --git a/examples/industrial_data_pretraining/ct_transformer_streaming/demo.py b/examples/industrial_data_pretraining/ct_transformer_streaming/demo.py
index 5ef8381..081fd19 100644
--- a/examples/industrial_data_pretraining/ct_transformer_streaming/demo.py
+++ b/examples/industrial_data_pretraining/ct_transformer_streaming/demo.py
@@ -12,7 +12,7 @@
rec_result_all = "outputs: "
cache = {}
for vad in vads:
- rec_result = model(input=vad, cache=cache)
+ rec_result = model.generate(input=vad, cache=cache)
print(rec_result)
rec_result_all += rec_result[0]['text']
--
Gitblit v1.9.1