matthoffner commited on
Commit
ebbbded
1 Parent(s): 4efb547

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -66,7 +66,7 @@ async def chat(request: ChatCompletionRequest):
66
  tokens = llm.tokenize(combined_messages)
67
 
68
  try:
69
- chat_chunks = llm.generate(tokens, max_new_tokens=request.max_tokens)
70
  except Exception as e:
71
  raise HTTPException(status_code=500, detail=str(e))
72
 
 
66
  tokens = llm.tokenize(combined_messages)
67
 
68
  try:
69
+ chat_chunks = llm.generate(tokens)
70
  except Exception as e:
71
  raise HTTPException(status_code=500, detail=str(e))
72