From 4e0404e04ed890717ead276e52c927a820326ec1 Mon Sep 17 00:00:00 2001
From: aky15 <ankeyu.aky@11.17.44.249>
Date: 星期三, 01 十一月 2023 16:47:13 +0800
Subject: [PATCH] fix rwkv infer bugs
---
funasr/bin/lm_inference_launch.py | 1 +
1 files changed, 1 insertions(+), 0 deletions(-)
diff --git a/funasr/bin/lm_inference_launch.py b/funasr/bin/lm_inference_launch.py
index 236a923..f12f50a 100644
--- a/funasr/bin/lm_inference_launch.py
+++ b/funasr/bin/lm_inference_launch.py
@@ -183,6 +183,7 @@
dtype=dtype,
batch_size=batch_size,
key_file=key_file,
+ preprocess_fn=preprocessor,
num_workers=num_workers,
)
--
Gitblit v1.9.1