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,