log new frames count

This commit is contained in:
yuvraj108c
2024-10-04 13:36:16 +04:00
parent 987334d48d
commit 08fa66cd1e
2 changed files with 3 additions and 2 deletions
+2 -2
View File
@@ -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,
+1
View File
@@ -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