losyer8 commited on
Commit
86b2954
1 Parent(s): 7b16214

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -62,9 +62,9 @@ tokenized_input = tokenizer.encode(text, add_special_tokens=False, return_tensor
62
  with torch.no_grad():
63
  output = model.generate(
64
  tokenized_input,
65
- max_new_tokens=100,
66
  do_sample=True,
67
- top_p=0.95,
68
  temperature=0.7,
69
  )[0]
70
  print(tokenizer.decode(output))
 
62
  with torch.no_grad():
63
  output = model.generate(
64
  tokenized_input,
65
+ max_new_tokens=20,
66
  do_sample=True,
67
+ top_p=0.90,
68
  temperature=0.7,
69
  )[0]
70
  print(tokenizer.decode(output))