Fix stop str merging (#225)

Co-authored-by: Enrique Shockwave <33002121+qeternity@users.noreply.github.com>

Fix stop str merging (#225)
Co-authored-by: Enrique Shockwave <33002121+qeternity@users.noreply.github.com>
c4e9ebe3 · Liangsheng Yin · GitHub · 3c2c5869 · c4e9ebe3
Unverified Commit c4e9ebe3 authored Feb 24, 2024 by Liangsheng Yin Committed by GitHub Feb 24, 2024
Show whitespace changes
Inline Side-by-side

Showing with 4 additions and 0 deletions

python/sglang/lang/interpreter.py python/sglang/lang/interpreter.py +4 -0

No files found.
--- a/python/sglang/lang/interpreter.py
+++ b/python/sglang/lang/interpreter.py
@@ -583,6 +583,10 @@ class StreamExecutor:
        if self.chat_template.stop_str:
            if not clone:
                clone = self.default_sampling_para.clone()
+            if clone.stop == ():
+                clone.stop = []
+            elif isinstance(clone.stop, str):
+                clone.stop = [clone.stop]
            clone.stop += self.chat_template.stop_str

        return clone or self.default_sampling_para