From b9824fab893843dde85a21a77faadc9d9ec0af34 Mon Sep 17 00:00:00 2001
From: x54-729 <45304952+x54-729@users.noreply.github.com>
Date: Tue, 19 Sep 2023 14:18:02 +0800
Subject: [PATCH] fix(tools): fix yield bug in stream_chat (#315)

---
 tools/transformers/modeling_internlm.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tools/transformers/modeling_internlm.py b/tools/transformers/modeling_internlm.py
index da7aaa0..5439ba7 100644
--- a/tools/transformers/modeling_internlm.py
+++ b/tools/transformers/modeling_internlm.py
@@ -869,7 +869,7 @@ class InternLMForCausalLM(InternLMPreTrainedModel):
             producer.start()
             while True:
                 res = response_queue.get()
-                if res is not None:
+                if res is None:
                     return
                 yield res