Update app.py
Browse files
app.py
CHANGED
@@ -16,7 +16,7 @@ def format_prompt(message, history):
|
|
16 |
return prompt
|
17 |
|
18 |
def generate(
|
19 |
-
prompt, history, system_prompt, max_new_tokens=100, temperature=0.7, repetition_penalty=1.2, top_p=0.95, top_k=1, random_seed=False,
|
20 |
):
|
21 |
temperature = float(temperature)
|
22 |
if temperature < 1e-2:
|
@@ -25,8 +25,6 @@ def generate(
|
|
25 |
|
26 |
if random_seed:
|
27 |
seed = random.randint(1, 100000)
|
28 |
-
else:
|
29 |
-
seed = manual_seed
|
30 |
|
31 |
generate_kwargs = dict(
|
32 |
temperature=temperature,
|
|
|
16 |
return prompt
|
17 |
|
18 |
def generate(
|
19 |
+
prompt, history, system_prompt, max_new_tokens=100, temperature=0.7, repetition_penalty=1.2, top_p=0.95, top_k=1, random_seed=False, seed=42,
|
20 |
):
|
21 |
temperature = float(temperature)
|
22 |
if temperature < 1e-2:
|
|
|
25 |
|
26 |
if random_seed:
|
27 |
seed = random.randint(1, 100000)
|
|
|
|
|
28 |
|
29 |
generate_kwargs = dict(
|
30 |
temperature=temperature,
|