fix remaning files affected by basicsr deprication

viking1304 · viking1304 · commit e7f06257f893 · 2024-03-11T16:56:52.000Z
diff --git a/scripts/deforum_helpers/frame_interpolation.py b/scripts/deforum_helpers/frame_interpolation.py
@@ -190,7 +190,11 @@ def prepare_film_inference(deforum_models_path, x_am, sl_enabled, sl_am, keep_im
     return interp_vid_path
 
 def check_and_download_film_model(model_name, model_dest_folder):
-    from basicsr.utils.download_util import load_file_from_url
+    try:
+        from modules.modelloader import load_file_from_url
+    except:
+        print("Try to fallback to basicsr with older modules")
+        from basicsr.utils.download_util import load_file_from_url
     if model_name == 'film_net_fp16.pt':
         model_dest_path = os.path.join(model_dest_folder, model_name)
         download_url = 'https://github.com/hithereai/frame-interpolation-pytorch/releases/download/film_net_fp16.pt/film_net_fp16.pt'
diff --git a/scripts/deforum_helpers/src/rife/rife_new_gen/RIFE_HDv3.py b/scripts/deforum_helpers/src/rife/rife_new_gen/RIFE_HDv3.py
@@ -102,7 +102,11 @@ def download_rife_model(path, deforum_models_path):
         target_file = f"{path}.pkl"
         target_path = os.path.join(deforum_models_path, target_file)
         if not os.path.exists(target_path):
-            from basicsr.utils.download_util import load_file_from_url
+            try:
+                from modules.modelloader import load_file_from_url
+            except:
+                print("Try to fallback to basicsr with older modules")
+                from basicsr.utils.download_util import load_file_from_url
             load_file_from_url(options[path][1], deforum_models_path)
             if checksum(target_path) != options[path][0]:
                 raise Exception(f"Error while downloading {target_file}. Please download from here: {options[path][1]} and place in: " + deforum_models_path)
