removing softmax?

interfaces/cap_minor_media.py

@@ -41,18 +41,8 @@ major_to_minor_map = dict(major_to_minor_map)
 
 
 def normalize_probs(probs: dict) -> dict:
-    keys, values = zip(*probs.items())
-    values = np.array(values)
-
-    # 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)))
+    total = sum(probs.values())
+    return {k: v / total for k, v in probs.items()}
 
 
 def check_huggingface_path(checkpoint_path: str):