mjbuehler commited on
Commit
6b8db6f
·
1 Parent(s): 80a21be

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -37,7 +37,7 @@ def generate_response (text_input="Biological materials offer amazing",
37
  top_k = top_k,
38
  top_p =top_p,
39
  num_return_sequences = num_return_sequences, eos_token_id=eos_token_id,
40
- do_sample =True,#skip_prompt=True,
41
  repetition_penalty=repetition_penalty,
42
  )
43
  return tokenizer.batch_decode(outputs[:,inputs.shape[1]:].detach().cpu().numpy(), skip_special_tokens=True)
 
37
  top_k = top_k,
38
  top_p =top_p,
39
  num_return_sequences = num_return_sequences, eos_token_id=eos_token_id,
40
+ do_sample =True,
41
  repetition_penalty=repetition_penalty,
42
  )
43
  return tokenizer.batch_decode(outputs[:,inputs.shape[1]:].detach().cpu().numpy(), skip_special_tokens=True)