Sengxian commited on
Commit
b387589
1 Parent(s): aa0674a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
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