add log to normalization to prevent small values bug

interfaces/cap_minor_media.py

@@ -39,13 +39,20 @@ for code in CAP_MIN_CODES:
     major_to_minor_map[major_id].append(code)
 major_to_minor_map = dict(major_to_minor_map)
 
+
 def normalize_probs(probs: dict) -> dict:
     keys, values = zip(*probs.items())
     values = np.array(values)
-    exp_values = np.exp(values - np.max(values))  # For numerical stability
-    sum_exp = np.sum(exp_values)
-    normalized = exp_values / sum_exp
-    return dict(zip(keys, map(float, normalized)))
+
+    # log-transform to spread small values (add small constant to avoid log(0))
+    log_values = np.log(values + 1e-300)
+    
+    # log-sum-exp trick for stability
+    max_log = np.max(log_values)
+    exp_values = np.exp(log_values - max_log)
+    softmax = exp_values / exp_values.sum()
+
+    return dict(zip(keys, map(float, softmax)))
 
 
 def check_huggingface_path(checkpoint_path: str):