Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -23,7 +23,7 @@ try:
|
|
23 |
tokenizer=tokenizer,
|
24 |
torch_dtype=torch.float16,
|
25 |
num_return_sequences=1,
|
26 |
-
do_sample=
|
27 |
truncation=True,
|
28 |
max_new_tokens=128
|
29 |
)
|
@@ -44,7 +44,7 @@ def respond(message, history, system_message, max_tokens, temperature, top_p):
|
|
44 |
max_new_tokens=max_tokens,
|
45 |
temperature=temperature,
|
46 |
top_p=top_p,
|
47 |
-
do_sample=
|
48 |
pad_token_id=tokenizer.eos_token_id
|
49 |
)[0]["generated_text"]
|
50 |
|
|
|
23 |
tokenizer=tokenizer,
|
24 |
torch_dtype=torch.float16,
|
25 |
num_return_sequences=1,
|
26 |
+
do_sample=False,
|
27 |
truncation=True,
|
28 |
max_new_tokens=128
|
29 |
)
|
|
|
44 |
max_new_tokens=max_tokens,
|
45 |
temperature=temperature,
|
46 |
top_p=top_p,
|
47 |
+
do_sample=False,
|
48 |
pad_token_id=tokenizer.eos_token_id
|
49 |
)[0]["generated_text"]
|
50 |
|