rewgwrth commited on
Commit
311a3c8
·
1 Parent(s): 33aa6dc
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -46,7 +46,7 @@ async def generate_text(request: RequestData):
46
  try:
47
  output = generator(
48
  request.prompt,
49
- max_new_tokens=min(request.max_tokens, 100),
50
  do_sample=False,
51
  num_beams=1,
52
  temperature=0.7,
 
46
  try:
47
  output = generator(
48
  request.prompt,
49
+ max_new_tokens=min(request.max_tokens, 10000),
50
  do_sample=False,
51
  num_beams=1,
52
  temperature=0.7,