fix issue https://github.com/THUDM/ChatGLM3/issues/227 (#9)

- fix issue https://github.com/THUDM/ChatGLM3/issues/227 (9c01557f168549c2f022b96904335f0e21ecb11d)


Co-authored-by: y A n g <[email protected]>

modeling_chatglm.py

@@ -1000,7 +1000,10 @@ class ChatGLMForConditionalGeneration(ChatGLMPreTrainedModel):
         content = ""
         history = deepcopy(history)
         for response in output.split("<|assistant|>"):
-            metadata, content = response.split("\n", maxsplit=1)
+            if "\n" in response:
+                metadata, content = response.split("\n", maxsplit=1)
+            else:
+                metadata, content = "", response
             if not metadata.strip():
                 content = content.strip()
                 history.append({"role": "assistant", "metadata": metadata, "content": content})