nielsr HF staff commited on
Commit
7aaadac
1 Parent(s): b43222a

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -0
README.md CHANGED
@@ -51,6 +51,7 @@ outputs = model.generate(
51
  length_penalty=1.0,
52
  temperature=1,
53
  )
 
54
  generated_text = processor.batch_decode(outputs, skip_special_tokens=True)[0].strip()
55
  print(generated_text)
56
  ```
 
51
  length_penalty=1.0,
52
  temperature=1,
53
  )
54
+ outputs[outputs == 0] = 2
55
  generated_text = processor.batch_decode(outputs, skip_special_tokens=True)[0].strip()
56
  print(generated_text)
57
  ```