update app
Browse files
app.py
CHANGED
@@ -42,7 +42,6 @@ def generate_response(message, temperature=0.7, max_new_tokens=256):
|
|
42 |
do_sample=True,
|
43 |
top_k=50,
|
44 |
top_p=0.9,
|
45 |
-
repetition_penalty=1.2,
|
46 |
pad_token_id=tokenizer.pad_token_id,
|
47 |
eos_token_id=tokenizer.eos_token_id,
|
48 |
batch_size=1
|
|
|
42 |
do_sample=True,
|
43 |
top_k=50,
|
44 |
top_p=0.9,
|
|
|
45 |
pad_token_id=tokenizer.pad_token_id,
|
46 |
eos_token_id=tokenizer.eos_token_id,
|
47 |
batch_size=1
|