From 26152e1343d821884c0a0344cf50bb6fcb027c92 Mon Sep 17 00:00:00 2001 From: Extraltodeus Date: Sat, 8 Jun 2024 02:13:44 +0200 Subject: [PATCH] adapted formulas --- nodes.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/nodes.py b/nodes.py index 122cf35..16fcf55 100644 --- a/nodes.py +++ b/nodes.py @@ -43,10 +43,10 @@ class temperature_patcher(): if self.eval_string != "": out = eval(self.eval_string) else: - if self.model_name == "SD1": - out *= math.log(q.size(-2), ldim) - 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 = ( out.transpose(1, 2).reshape(b, -1, heads * dim_head)