akiko19191 commited on
Commit
cc63503
·
verified ·
1 Parent(s): ad2ec99

Update utils/llms.py

Browse files
Files changed (1) hide show
  1. utils/llms.py +1 -1
utils/llms.py CHANGED
@@ -93,7 +93,7 @@ def gpt4stream(messages,model,api_keys):
93
  helper.q.put_nowait("RESULT: "+cunk)
94
  else:
95
  helper.q.put_nowait("END")
96
- elif model=="qwen-qwq-32b":
97
  os.environ["GROQ_API_KEY"] =random.choice(groq_api_keys)
98
  response = completion(model="groq/qwen-qwq-32b", messages=messages, stream=True)
99
 
 
93
  helper.q.put_nowait("RESULT: "+cunk)
94
  else:
95
  helper.q.put_nowait("END")
96
+ elif model=="gemini-2.0-pro-exp-02-05":
97
  os.environ["GROQ_API_KEY"] =random.choice(groq_api_keys)
98
  response = completion(model="groq/qwen-qwq-32b", messages=messages, stream=True)
99