From 142e30622e6e915a46230c978b82325d65794b19 Mon Sep 17 00:00:00 2001 From: josc146 Date: Mon, 29 May 2023 20:17:29 +0800 Subject: [PATCH] send response even token is END_OF_TEXT --- backend-python/utils/rwkv.py | 1 + 1 file changed, 1 insertion(+) diff --git a/backend-python/utils/rwkv.py b/backend-python/utils/rwkv.py index 49f5364..48e9414 100644 --- a/backend-python/utils/rwkv.py +++ b/backend-python/utils/rwkv.py @@ -138,6 +138,7 @@ The following is a coherent verbose detailed conversation between a girl named { ) if token == END_OF_TEXT: + yield response, "" break if token not in occurrence: occurrence[token] = 1