Spaces:
Runtime error
Runtime error
Update main.py
Browse files
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.
|
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.
|
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 |
|