fix: use dtype-preserving cast in SGM-VFI softsplat fallback
Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
This commit is contained in:
1 parent
8d8407ec9d
commit
5ce7b0edcb
1 file changed
+2
-2
@@ -258,8 +258,8 @@ def _pytorch_softsplat(tenIn, tenFlow):
|
||||
|
||||
nw_x = flt_x.floor().long()
|
||||
nw_y = flt_y.floor().long()
|
||||
frac_x = flt_x - nw_x.float()
|
||||
frac_y = flt_y - nw_y.float()
|
||||
frac_x = flt_x - nw_x.to(flt_x.dtype)
|
||||
frac_y = flt_y - nw_y.to(flt_y.dtype)
|
||||
|
||||
w_nw = (1.0 - frac_x) * (1.0 - frac_y) * valid
|
||||
w_ne = frac_x * (1.0 - frac_y) * valid
|
||||
|
||||
Reference in new issue
Block a user