Update ace_plus_fft_node.py

2025-04-21 10:51:53 +08:00
parent ff0542a85f
commit 6067db7e7a
1 changed files with 1 additions and 1 deletions
@@ -379,7 +379,7 @@ class AcePlusFFTProcessor:
            slice_w = 0

        H, W = edit_image.shape[-2:]
-        scale = min(1.0, math.sqrt(self.max_seq_len * 2 / ((H / self.d) * (W / self.d))))
+        scale = min(1.0, math.sqrt(self.max_seq_len / ((H / self.d) * (W / self.d))))
        rH = int(H * scale) // self.d * self.d
        rW = int(W * scale) // self.d * self.d
        slice_w = int(slice_w * scale) // self.d * self.d