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 |
|