Skip top_p computations when set to 1.0 (#8905)

Signed-off-by: Olivier Delalleau <507137+odelalleau@users.noreply.github.com>
Co-authored-by: Pablo Garay <palenq@gmail.com>

nemo/collections/nlp/modules/common/text_generation_utils.py

@@ -303,7 +303,7 @@ def top_k_logits(logits, top_k=0, top_p=0.0, filter_value=-float('Inf'), started
         else:
             logits[indices_to_remove] = filter_value
 
-    if top_p > 0.0:
+    if 0.0 < top_p < 1.0:
         # Cconvert to 1D
         sorted_logits, sorted_indices = torch.sort(logits, descending=True, dim=-1)
         cumulative_probs = torch.cumsum(F.softmax(sorted_logits, dim=-1), dim=-1)