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