feat: ccvfi

Author: Tohrusky

cccv/config/vfi/rife_config.py

@@ -8,7 +8,7 @@
 class RIFEConfig(VFIBaseConfig):
     arch: Union[ArchType, str] = ArchType.IFNET
     model: Union[ModelType, str] = ModelType.RIFE
-    num_frame: int = 3
+    num_frame: int = 2
 
 
 RIFEConfigs = [

example/vfi_vs.py

@@ -10,10 +10,7 @@
 
 # --- IFNet, use fp16 to inference (vs.RGBH)
 
-model: VFIBaseModel = AutoModel.from_pretrained(
-    ConfigType.RIFE_IFNet_v426_heavy,
-    fp16=True,
-)
+model: VFIBaseModel = AutoModel.from_pretrained(ConfigType.RIFE_IFNet_v426_heavy, fp16=True, tile=None)
 
 core.num_threads = 1  # should be set to single thread now, TODO: fix it
 clip = core.bs.VideoSource(source="s.mkv")

example/vsr.py

@@ -7,7 +7,7 @@
 img = cv2.imdecode(np.fromfile("../assets/test.jpg", dtype=np.uint8), cv2.IMREAD_COLOR)
 imgList = [img, img, img]
 
-model: VSRBaseModel = AutoModel.from_pretrained(ConfigType.AnimeSR_v2_4x, fp16=False)
+model: VSRBaseModel = AutoModel.from_pretrained(ConfigType.AnimeSR_v2_4x)
 
 imgOutList = model.inference_image_list(imgList)