Add falback to basicr for legacy versions

viking1304 · web-flow · commit 7c6cf6747421 · 2024-03-08T02:06:28.000+01:00
diff --git a/scripts/deforum_helpers/general_utils.py b/scripts/deforum_helpers/general_utils.py
@@ -18,7 +18,11 @@
 import shutil
 import hashlib
 from modules.shared import opts
-from modules.modelloader 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
 
 def debug_print(message):
     DEBUG_MODE = opts.data.get("deforum_debug_mode_enabled", False)
diff --git a/scripts/deforum_helpers/upscaling.py b/scripts/deforum_helpers/upscaling.py
@@ -22,11 +22,16 @@
 from .frame_interpolation import clean_folder_name
 from .general_utils import duplicate_pngs_from_folder, checksum
 from .video_audio_utilities import vid2frames, ffmpeg_stitch_video, extract_number, media_file_has_audio
-from modules.modelloader import load_file_from_url
 from .rich import console
 
 from modules.shared import opts
 
+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
+
 # NCNN Upscale section START
 def process_ncnn_upscale_vid_upload_logic(vid_path, in_vid_fps, in_vid_res, out_vid_res, models_path, upscale_model, upscale_factor, keep_imgs, f_location, f_crf, f_preset, current_user_os):
     print(f"Got a request to *upscale* a video using {upscale_model} at {upscale_factor}")
diff --git a/scripts/deforum_helpers/video_audio_utilities.py b/scripts/deforum_helpers/video_audio_utilities.py
@@ -29,10 +29,14 @@
 from pkg_resources import resource_filename
 from modules.shared import state, opts
 from .general_utils import checksum, clean_gradio_path_strings, debug_print
-from modules.modelloader import load_file_from_url
 from .rich import console
 import shutil
 from threading import Thread
+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
 
 def convert_image(input_path, output_path):
     # Read the input image
