jonathanjordan21 commited on
Commit
ca5271c
1 Parent(s): 6e897aa

Update apis/chat_api.py

Browse files
Files changed (1) hide show
  1. apis/chat_api.py +4 -4
apis/chat_api.py CHANGED
@@ -200,11 +200,11 @@ class ChatAPIApp:
200
  streamer = HuggingfaceStreamer(model=item.model)
201
  stream_response = streamer.chat_response(
202
  prompt=item.prompt,
203
- temperature=item.options.get('temperature'),
204
- top_p=item.options.get('top_p'),
205
- max_new_tokens=item.options.get('max_new_tokens'),
206
  api_key=api_key,
207
- use_cache=item.options.get('use_cache'),
208
  )
209
 
210
  if item.stream:
 
200
  streamer = HuggingfaceStreamer(model=item.model)
201
  stream_response = streamer.chat_response(
202
  prompt=item.prompt,
203
+ temperature=item.options.get('temperature', 0.6),
204
+ top_p=item.options.get('top_p', 0.95),
205
+ max_new_tokens=item.options.get('max_new_tokens', -1),
206
  api_key=api_key,
207
+ use_cache=item.options.get('use_cache', False),
208
  )
209
 
210
  if item.stream: