diff --git a/nodes.py b/nodes.py
index 16fcf55..b39a506 100644
--- a/nodes.py
+++ b/nodes.py
@@ -43,10 +43,7 @@ class temperature_patcher():
             if self.eval_string != "":
                 out  = eval(self.eval_string)
             else:
-                if self.model_name == "SDXL":
-                    out *= math.log(q.size(-2)*(q.size(-1)**2/ldim), ldim)
-                else:
-                    out *= math.log(q.size(-2), ldim)
+                out *= math.log(q.size(-2), ldim)
 
         out = (
             out.transpose(1, 2).reshape(b, -1, heads * dim_head)