Respect user's pyr_level setting at all resolutions

Previously the user's pyr_level was overridden for >=540p content.
Now the setting is always used, with the tooltip recommending values
per resolution instead.

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
This commit is contained in:
2026-02-12 18:50:29 +01:00
parent ffde07a89a
commit 4e6f9eb896
2 changed files with 2 additions and 12 deletions

View File

@@ -58,22 +58,12 @@ class BiMVFIModel:
img0 = frame0.to(device)
img1 = frame1.to(device)
_, _, h, w = img0.shape
if h >= 2160:
pyr_level = 7
elif h >= 1080:
pyr_level = 6
elif h >= 540:
pyr_level = 5
else:
pyr_level = self.pyr_level
time_step_tensor = torch.tensor([time_step], device=device).view(1, 1, 1, 1)
result_dict = self.model(
img0=img0, img1=img1,
time_step=time_step_tensor,
pyr_level=pyr_level,
pyr_level=self.pyr_level,
)
interp = result_dict["imgt_pred"]