Spaces:
Paused
Paused
Danielrahmai1991
commited on
Commit
•
97e04f5
1
Parent(s):
dc1fe45
Update app.py
Browse files
app.py
CHANGED
@@ -66,7 +66,7 @@ def llm_run(prompt, max_length, top_p, temprature, top_k, messages):
|
|
66 |
)
|
67 |
generate_kwargs = dict(
|
68 |
max_length=int(max_length),top_p=float(top_p), do_sample=True,
|
69 |
-
top_k=int(top_k), streamer=streamer, temperature=
|
70 |
)
|
71 |
|
72 |
t = Thread(target=model.generate, args=(input_ids,), kwargs=generate_kwargs)
|
|
|
66 |
)
|
67 |
generate_kwargs = dict(
|
68 |
max_length=int(max_length),top_p=float(top_p), do_sample=True,
|
69 |
+
top_k=int(top_k), streamer=streamer, temperature=float(temprature), repetition_penalty=1.2
|
70 |
)
|
71 |
|
72 |
t = Thread(target=model.generate, args=(input_ids,), kwargs=generate_kwargs)
|