update app
Browse files
app.py
CHANGED
|
@@ -80,14 +80,14 @@ def generate_stream(prompt, max_new_tokens=128):
|
|
| 80 |
model.generate(
|
| 81 |
**inputs,
|
| 82 |
max_new_tokens=max_new_tokens,
|
| 83 |
-
min_new_tokens=16,
|
| 84 |
do_sample=True,
|
| 85 |
temperature=0.7,
|
| 86 |
top_p=0.9,
|
| 87 |
streamer=streamer,
|
| 88 |
-
early_stopping=True,
|
| 89 |
-
eos_token_id=tokenizer.eos_token_id,
|
| 90 |
-
pad_token_id=tokenizer.pad_token_id,
|
| 91 |
use_cache=True,
|
| 92 |
)
|
| 93 |
except Exception as e:
|
|
|
|
| 80 |
model.generate(
|
| 81 |
**inputs,
|
| 82 |
max_new_tokens=max_new_tokens,
|
| 83 |
+
# min_new_tokens=16,
|
| 84 |
do_sample=True,
|
| 85 |
temperature=0.7,
|
| 86 |
top_p=0.9,
|
| 87 |
streamer=streamer,
|
| 88 |
+
# early_stopping=True,
|
| 89 |
+
# eos_token_id=tokenizer.eos_token_id,
|
| 90 |
+
# pad_token_id=tokenizer.pad_token_id,
|
| 91 |
use_cache=True,
|
| 92 |
)
|
| 93 |
except Exception as e:
|