fix: typecast bf16 to float

Author: molereddy

src/evals/metrics/mia/min_k.py

@@ -17,7 +17,7 @@ def compute_batch_values(self, batch):
 
     def compute_score(self, sample_stats):
         """Score single sample using min-k negative log probs scores attack."""
-        lp = sample_stats.cpu().numpy()
+        lp = sample_stats.float().cpu().numpy()
         if lp.size == 0:
             return 0
 

src/evals/metrics/mia/min_k_plus_plus.py

@@ -30,7 +30,7 @@ def compute_score(self, sample_stats):
 
         # Handle numerical stability
         sigma = torch.clamp(sigma, min=1e-6)
-        scores = (target_prob.cpu().numpy() - mu.cpu().numpy()) / torch.sqrt(
+        scores = (target_prob.float().cpu().numpy() - mu.float().cpu().numpy()) / torch.sqrt(
             sigma
         ).cpu().numpy()