akiko19191 commited on
Commit
ccb8dba
·
verified ·
1 Parent(s): 1cc1b0a

Update utils/llms.py

Browse files
Files changed (1) hide show
  1. utils/llms.py +1 -1
utils/llms.py CHANGED
@@ -125,7 +125,7 @@ def chatstream(messages,model,api_keys):
125
  yield("RESULT: "+cunk)
126
  elif model in chutes_models:
127
  response = completion(model=f"openai/{chutes_models[model]}",api_key=chutes_key,base_url="https://llm.chutes.ai/v1", messages=messages, stream=True)
128
- if model == "MAI-DS-R1-FP8" or model == "GLM-Z1-32B-0414":
129
 
130
  yield("<think> \n")
131
 
 
125
  yield("RESULT: "+cunk)
126
  elif model in chutes_models:
127
  response = completion(model=f"openai/{chutes_models[model]}",api_key=chutes_key,base_url="https://llm.chutes.ai/v1", messages=messages, stream=True)
128
+ if model == "MAI-DS-R1-FP8" or model == "GLM-Z1-32B-0414" or model == "DeepSeek-R1T-Chimera" or model == "QwQ-32B-ArliAI-RpR-v1":
129
 
130
  yield("<think> \n")
131