Re-add do_sample=false option.