From 08fa66cd1ea7f293a3369fbbfe580c349c942655 Mon Sep 17 00:00:00 2001
From: yuvraj108c <yuvraj108c@gmail.com>
Date: Fri, 4 Oct 2024 13:36:16 +0400
Subject: [PATCH] log new frames count

---
 __init__.py      | 4 ++--
 vfi_utilities.py | 1 +
 2 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/__init__.py b/__init__.py
index 44dba2c..c13155b 100644
--- a/__init__.py
+++ b/__init__.py
@@ -16,7 +16,7 @@ class RifeTensorrt:
             "required": {
                 "frames": ("IMAGE", ),
                 "engine": (os.listdir(ENGINE_DIR),),
-                "clear_cache_after_n_frames": ("INT", {"default": 50, "min": 1, "max": 1000}),
+                "clear_cache_after_n_frames": ("INT", {"default": 100, "min": 1, "max": 1000}),
                 "multiplier": ("INT", {"default": 2, "min": 1}),
                 "use_cuda_graph": ("BOOLEAN", {"default": True}),
                 "keep_model_loaded": ("BOOLEAN", {"default": False}),
@@ -32,7 +32,7 @@ class RifeTensorrt:
         self,
         frames,
         engine,
-        clear_cache_after_n_frames=50,
+        clear_cache_after_n_frames=100,
         multiplier=2,
         use_cuda_graph=True,
         keep_model_loaded=False,
diff --git a/vfi_utilities.py b/vfi_utilities.py
index 37aaa31..f9ec616 100644
--- a/vfi_utilities.py
+++ b/vfi_utilities.py
@@ -74,6 +74,7 @@ def generate_frames_rife(
 
     # Append final frame
     output_frames[out_len] = frames[-1:]
+    logger(f"done! - {(len(frames) -1) * (multiplier-1)} new frames generated at resolution: {output_frames[0].shape}")
     out_len += 1
 
     # clear cache for courtesy