Update README.md

README.md

@@ -50,7 +50,7 @@ max_length = 80               # the maximum number of tokens for the output of t
 repitition_penalty = 1.2      # the penalty value for each repetition of a token
 num_return_sequences=5        # the number of results to generate
 
-# generate the result with contrastive search. generate 5 results with the highest probability out of 10.
+# generate the result with contrastive search
 input_ids = tokenizer(prompt, return_tensors='pt').input_ids
 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, penalty_alpha=0.6, no_repeat_ngram_size=1, early_stopping=True)