allow groundtruth mismatch in AcceptanceRateValidation

Signed-off-by: Ye Yu <[email protected]>

Author: yeyu-nvidia

modelopt/torch/speculative/utils.py

@@ -331,7 +331,9 @@ def validate(
 
         if ground_truth is None:
             ground_truth = self.get_ground_truth(input_ids, osl)
-        ground_truth = self.check_data_consistency_across_ranks(ground_truth)
+        ground_truth = self.check_data_consistency_across_ranks(
+            ground_truth, fail_when_mismatch=False
+        )
 
         cnt = 0
         draft_tokens = None