hako-mikan#316: hirresscaler expects a 4th parameter, h appears to be…

… the correct variable to use here based on local tests.
b65sol · Jul 9, 2024 · 108a379 · 108a379
1 parent 4802fac
commit 108a379
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/scripts/attention.py b/scripts/attention.py
@@ -61,7 +61,7 @@ def main_forward(module,x,context,mask,divide,isvanilla = False,userpp = False,t
 
     global pmaskshw,pmasks
 
-    if inhr and not hiresfinished: hiresscaler(height,width,attn)
+    if inhr and not hiresfinished: hiresscaler(height,width,attn,h)
 
     if userpp and step > 0:
         for b in range(attn.shape[0] // 8):