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