Update app.py
Browse files
app.py
CHANGED
@@ -81,11 +81,12 @@ system_instructions1 = "[SYSTEM] Sasha is a friendly and enthusiastic person, bo
|
|
81 |
|
82 |
async def generate1(prompt):
|
83 |
generate_kwargs = dict(
|
84 |
-
temperature=0
|
85 |
-
max_new_tokens=512,
|
86 |
top_p=0.95,
|
87 |
-
repetition_penalty=
|
88 |
-
do_sample=
|
|
|
89 |
)
|
90 |
formatted_prompt = system_instructions1 + prompt + "[SASHA]"
|
91 |
stream = client1.text_generation(
|
|
|
81 |
|
82 |
async def generate1(prompt):
|
83 |
generate_kwargs = dict(
|
84 |
+
temperature=1.0,
|
85 |
+
max_new_tokens=128, #def 512 very long sometimes, 256 semi large
|
86 |
top_p=0.95,
|
87 |
+
repetition_penalty=0.9,
|
88 |
+
do_sample=True,
|
89 |
+
seed=42,
|
90 |
)
|
91 |
formatted_prompt = system_instructions1 + prompt + "[SASHA]"
|
92 |
stream = client1.text_generation(
|