xu song commited on
Commit
55f05a7
·
1 Parent(s): ed3e00b
Files changed (1) hide show
  1. models/cpp_qwen2.py +2 -2
models/cpp_qwen2.py CHANGED
@@ -142,11 +142,11 @@ if __name__ == "__main__":
142
 
143
  message = {"role": "system", "content": "你是一个导游。"}
144
  print(message)
145
- for generated_text, all_tokens in bot.generate_query(message, stream=True):
146
  print(generated_text, all_tokens)
147
 
148
  message = {"role": "user", "content": generated_text}
149
  print(message)
150
- for generated_text, all_tokens in bot.generate_query(message, stream=True):
151
  print(generated_text, all_tokens)
152
 
 
142
 
143
  message = {"role": "system", "content": "你是一个导游。"}
144
  print(message)
145
+ for generated_text, all_tokens in bot.generate_query(message, [], stream=True):
146
  print(generated_text, all_tokens)
147
 
148
  message = {"role": "user", "content": generated_text}
149
  print(message)
150
+ for generated_text, all_tokens in bot.generate_query(message, all_tokens, stream=True):
151
  print(generated_text, all_tokens)
152