FredZhang7 commited on
Commit
d2e6900
1 Parent(s): d6fb869

Fixed examples

Browse files
Files changed (1) hide show
  1. README.md +1 -2
README.md CHANGED
@@ -82,9 +82,8 @@ for i in range(len(output)):
82
  No comma style:
83
  ![constrastive search](./constrastive_search.png)
84
 
85
- <br>
86
 
87
- To bring back the commas, assign output without penalty_alpha:
88
  ```python
89
  output = model.generate(input_ids, do_sample=True, temperature=temperature, top_k=top_k, max_length=max_length, num_return_sequences=num_return_sequences, repetition_penalty=repitition_penalty, no_repeat_ngram_size=1, early_stopping=True)
90
  ```
 
82
  No comma style:
83
  ![constrastive search](./constrastive_search.png)
84
 
 
85
 
86
+ To bring back the commas, assign output without `penalty_alpha`:
87
  ```python
88
  output = model.generate(input_ids, do_sample=True, temperature=temperature, top_k=top_k, max_length=max_length, num_return_sequences=num_return_sequences, repetition_penalty=repitition_penalty, no_repeat_ngram_size=1, early_stopping=True)
89
  ```