From 9f68ea042e126b922a48f2b774741f93dffdb534 Mon Sep 17 00:00:00 2001 From: er1cw00 Date: Sat, 29 Aug 2026 00:29:38 +0800 Subject: [PATCH] misc: add time elapsed --- rife_node.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/rife_node.py b/rife_node.py index b84437c..880224c 100644 --- a/rife_node.py +++ b/rife_node.py @@ -5,6 +5,7 @@ import pathlib import einops import folder_paths import typing +import time import urllib.request import urllib.error #from vfi_utils import load_file_from_github_release @@ -160,7 +161,7 @@ class RIFE_VFI: Returns: tuple: A tuple containing the output interpolated frames. """ - + start_time = time.perf_counter() model_path = self._get_model_path(ckpt_name) arch_ver = CKPT_NAME_VER_DICT[ckpt_name] @@ -285,4 +286,8 @@ class RIFE_VFI: # Always return float32 — numpy and all downstream ComfyUI nodes require it out_tensor = torch.cat(output_frames, dim=0).to(torch.float32) + + stop_time = time.perf_counter() + elapsed_time = stop_time - start_time + print(f"VFI elapsed: {elapsed_time:.4f} s ({elapsed_time * 1000:.2f} ms)") return (postprocess_frames(out_tensor),)