Upload main.py
Browse filespoprawka do przecinka
main.py
CHANGED
@@ -68,7 +68,7 @@ async def generate_text(request: GenerationRequest):
|
|
68 |
max_new_tokens=request.max_new_tokens,
|
69 |
temperature=request.temperature,
|
70 |
top_p=request.top_p,
|
71 |
-
do_sample=True
|
72 |
)
|
73 |
# Pipeline zwraca list臋 s艂ownik贸w, bierzemy pierwszy wynik
|
74 |
return {"generated_text": generated_text[0]["generated_text"]}
|
|
|
68 |
max_new_tokens=request.max_new_tokens,
|
69 |
temperature=request.temperature,
|
70 |
top_p=request.top_p,
|
71 |
+
do_sample=True # Wa偶ne dla generowania z temperatur膮
|
72 |
)
|
73 |
# Pipeline zwraca list臋 s艂ownik贸w, bierzemy pierwszy wynik
|
74 |
return {"generated_text": generated_text[0]["generated_text"]}
|