Ethanfel/ComfyUI-Tween
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: use dtype-preserving cast in SGM-VFI softsplat fallback

Browse Source 
Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
This commit is contained in:
Ethanfel
2026-04-11 02:05:24 +02:00
parent 8d8407ec9d
commit 5ce7b0edcb
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
sgm_vfi_arch/softsplat.py
+2 -2
View File
@@ -258,8 +258,8 @@ def _pytorch_softsplat(tenIn, tenFlow):
nw_x = flt_x.floor().long() nw_x = flt_x.floor().long()
nw_y = flt_y.floor().long() nw_y = flt_y.floor().long()
frac_x = flt_x - nw_x.float() frac_x = flt_x - nw_x.to(flt_x.dtype)
frac_y = flt_y - nw_y.float() frac_y = flt_y - nw_y.to(flt_y.dtype)
w_nw = (1.0 - frac_x) * (1.0 - frac_y) * valid w_nw = (1.0 - frac_x) * (1.0 - frac_y) * valid
w_ne = frac_x * (1.0 - frac_y) * valid w_ne = frac_x * (1.0 - frac_y) * valid