Update README.md
Browse files
README.md
CHANGED
|
@@ -78,7 +78,7 @@ nfe, generated_tokens = model.generate_spd(
|
|
| 78 |
inputs=input_ids,
|
| 79 |
gen_length=2048,
|
| 80 |
block_length=32,
|
| 81 |
-
threshold=0.
|
| 82 |
)
|
| 83 |
|
| 84 |
generated_answer = tokenizer.decode(
|
|
|
|
| 78 |
inputs=input_ids,
|
| 79 |
gen_length=2048,
|
| 80 |
block_length=32,
|
| 81 |
+
threshold=0.5,
|
| 82 |
)
|
| 83 |
|
| 84 |
generated_answer = tokenizer.decode(
|