diff --git a/codeflash/process/infer.py b/codeflash/process/infer.py
index bbbaefb88..46423f685 100644
--- a/codeflash/process/infer.py
+++ b/codeflash/process/infer.py
@@ -2,7 +2,8 @@
 
 
 def sigmoid_stable(x):
-    return np.where(x >= 0, 1 / (1 + np.exp(-x)), np.exp(x) / (1 + np.exp(x)))
+    exp_x = np.exp(-np.abs(x))
+    return np.where(x >= 0, 1 / (1 + exp_x), exp_x / (1 + exp_x))
 
 
 def postprocess(logits: np.array, max_detections: int = 8):