diff --git a/codeflash/process/infer.py b/codeflash/process/infer.py
index bbbaefb88..b79717782 100644
--- a/codeflash/process/infer.py
+++ b/codeflash/process/infer.py
@@ -2,7 +2,10 @@
 
 
 def sigmoid_stable(x):
-    return np.where(x >= 0, 1 / (1 + np.exp(-x)), np.exp(x) / (1 + np.exp(x)))
+    exp_neg_x = np.exp(-x)
+    exp_x = np.exp(x)
+    # Use precomputed exponentials to avoid redundant calculation
+    return np.where(x >= 0, 1 / (1 + exp_neg_x), exp_x / (1 + exp_x))
 
 
 def postprocess(logits: np.array, max_detections: int = 8):