Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -19,12 +19,12 @@ def format_prompt(message, history):
|
|
19 |
return prompt
|
20 |
|
21 |
def generate(prompt,history,max_new_tokens,seed):
|
22 |
-
|
23 |
print(seed)
|
24 |
system_prompt = prompts.ASSISTANT
|
25 |
generate_kwargs = dict(
|
26 |
temperature=0.9,
|
27 |
-
max_new_tokens=
|
28 |
top_p=0.95,
|
29 |
repetition_penalty=1.0,
|
30 |
do_sample=True,
|
|
|
19 |
return prompt
|
20 |
|
21 |
def generate(prompt,history,max_new_tokens,seed):
|
22 |
+
seed = random.randint(1,9999999999999)
|
23 |
print(seed)
|
24 |
system_prompt = prompts.ASSISTANT
|
25 |
generate_kwargs = dict(
|
26 |
temperature=0.9,
|
27 |
+
max_new_tokens=512,
|
28 |
top_p=0.95,
|
29 |
repetition_penalty=1.0,
|
30 |
do_sample=True,
|