#26 fix upscaling error due to unhandled blank rife model

the-database · Aug 15, 2024 · 3e1f41b · 3e1f41b
1 parent d14059f
commit 3e1f41b
Showing 1 changed file with 11 additions and 3 deletions.
diff --git a/VideoJaNai/backend/animejanai/core/animejanai_config.py b/VideoJaNai/backend/animejanai/core/animejanai_config.py
@@ -46,6 +46,14 @@ def read_config_by_chain(flat_conf, section, chain, num_models):
     if max_fps == 0:
         max_fps = float("inf")
 
+    default_rife_model = 422
+
+    rife_model_str = flat_conf[section].get(f'chain_{chain}_rife_model', default_rife_model)
+    if not rife_model_str:
+        rife_model = default_rife_model
+    else:
+        rife_model = int(rife_model_str)
+
     return {
         'min_px': min_width * min_height,
         'max_px': max_width * max_height,
@@ -55,9 +63,9 @@ def read_config_by_chain(flat_conf, section, chain, num_models):
         'max_fps': max_fps,
         'models': [read_config_by_chain_model(flat_conf, section, chain, i) for i in range(1, num_models + 1)],
         'rife': parse_bool(flat_conf[section].get(f'chain_{chain}_rife', 'no')),
-        'rife_factor_numerator': int(float(flat_conf[section].get(f'chain_{chain}_rife_factor_numerator', 1))),
-        'rife_factor_denominator': int(float(flat_conf[section].get(f'chain_{chain}_rife_factor_denominator', 1))),
-        'rife_model': int(float(flat_conf[section].get(f'chain_{chain}_rife_model', 414))),
+        'rife_factor_numerator': int(flat_conf[section].get(f'chain_{chain}_rife_factor_numerator', 1)),
+        'rife_factor_denominator': int(flat_conf[section].get(f'chain_{chain}_rife_factor_denominator', 1)),
+        'rife_model': rife_model,
         'rife_ensemble': parse_bool(flat_conf[section].get(f'chain_{chain}_rife_ensemble', 'no')),
         'rife_scene_detect_threshold': float(flat_conf[section].get(f'chain_{chain}_rife_scene_detect_threshold', 0.150)),
         'final_resize_height': float(flat_conf[section].get(f'chain_{chain}_final_resize_height', 0)),