Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -33,6 +33,7 @@ def predict(text, seed, out_seq_length, min_gen_length, sampling_strategy,
|
|
| 33 |
"num_beams": num_beams,
|
| 34 |
"max_tokens": out_seq_length,
|
| 35 |
"no_repeat_ngram": no_repeat_ngram_size,
|
|
|
|
| 36 |
"seed": seed
|
| 37 |
})
|
| 38 |
|
|
|
|
| 33 |
"num_beams": num_beams,
|
| 34 |
"max_tokens": out_seq_length,
|
| 35 |
"no_repeat_ngram": no_repeat_ngram_size,
|
| 36 |
+
"quantization": "int4",
|
| 37 |
"seed": seed
|
| 38 |
})
|
| 39 |
|