Update modeling_qwen3_guard.py

modeling_qwen3_guard.py

@@ -626,6 +626,15 @@ class Qwen3ForGuardModel(Qwen3PreTrainedModel):
 
         return result, stream_state
 
+    @torch.no_grad()
+    def close_stream(self, stream_state: Optional[Generator]) -> None:
+        if stream_state is not None:
+            try:
+                stream_state.send(None)
+            except StopIteration:
+                pass
+            finally:
+                stream_state.close()
 
 __all__ = [
     "Qwen3PreTrainedModel",