MatteoScript commited on
Commit
49dd983
·
verified ·
1 Parent(s): b2f2237

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +2 -2
main.py CHANGED
@@ -60,7 +60,7 @@ def read_root(request: Request, input_data: InputData):
60
  generated_response = generate(input_text, history, temperature, max_new_tokens, top_p, repetition_penalty)
61
  return {"response": generated_response}
62
 
63
- def generate(prompt, history, temperature=0.2, max_new_tokens=30000, top_p=0.95, repetition_penalty=1.0):
64
  temperature = float(temperature)
65
  if temperature < 1e-2:
66
  temperature = 1e-2
@@ -116,7 +116,7 @@ async def GeneraTestoAsync(url, input_data):
116
  'top_p': input_data.top_p,
117
  'repetition_penalty': input_data.repetition_penalty
118
  }
119
- tasks.append(make_request(session, token, data, url) for _ in range(input_data.NumeroGenerazioni))
120
  #tasks = [make_request(session, token, data, url) for _ in range(input_data.NumeroGenerazioni)]
121
  return await asyncio.gather(*tasks)
122
 
 
60
  generated_response = generate(input_text, history, temperature, max_new_tokens, top_p, repetition_penalty)
61
  return {"response": generated_response}
62
 
63
+ def generate(prompt, history, temperature=0.7, max_new_tokens=30000, top_p=0.95, repetition_penalty=1.0):
64
  temperature = float(temperature)
65
  if temperature < 1e-2:
66
  temperature = 1e-2
 
116
  'top_p': input_data.top_p,
117
  'repetition_penalty': input_data.repetition_penalty
118
  }
119
+ tasks.extend([make_request(session, token, data, url) for _ in range(input_data.NumeroGenerazioni)])
120
  #tasks = [make_request(session, token, data, url) for _ in range(input_data.NumeroGenerazioni)]
121
  return await asyncio.gather(*tasks)
122