From eff2c826711de6c69a3ad8badb383aee270f117d Mon Sep 17 00:00:00 2001
From: hzwer <598460606@163.com>
Date: Fri, 25 Jul 2025 17:02:10 +0800
Subject: [PATCH] Update RIFE.py

---
 model/RIFE.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/model/RIFE.py b/model/RIFE.py
index 9af628a..d6e6c2c 100644
--- a/model/RIFE.py
+++ b/model/RIFE.py
@@ -53,7 +53,9 @@ class Model:
         if rank == 0:
             torch.save(self.flownet.state_dict(),'{}/flownet.pkl'.format(path))
 
-    def inference(self, img0, img1, scale=1, scale_list=[4, 2, 1], TTA=False, timestep=0.5):
+    def inference(self, img0, img1, scale=1, scale_list=None, TTA=False, timestep=0.5):
+        if scale_list is None:
+            scale_list = [4, 2, 1]
         for i in range(3):
             scale_list[i] = scale_list[i] * 1.0 / scale
         imgs = torch.cat((img0, img1), 1)