minor fix

Signed-off-by: dimapihtar <[email protected]>

Author: dimapihtar

megatron/training/datasets/fim_dataset.py

@@ -168,7 +168,6 @@ def _query_document_sample_shuffle_indices(self, idx: int) -> Tuple[np.ndarray,
     def _fim_permute_sequence(self, sequence, rate):
         return self._permute(
             sequence,
-            self.np_rng,
             rate,
             self.fim_spm_rate,
             self.config.tokenizer,