ComfyUI Rife TensorRT ⚡

This project provides a TensorRT implementation of RIFE for ultra fast frame interpolation inside ComfyUI

This project is licensed under CC BY-NC-SA, everyone is FREE to access, use, modify and redistribute with the same license.

If you like the project, please give me a star! ⭐

---

🚀 Installation

Navigate to the ComfyUI /custom_nodes directory

git clone https://github.com/yuvraj108c/ComfyUI-Rife-Tensorrt
cd ./ComfyUI-Rife-Tensorrt
pip install -r requirements.txt

🛠️ Building Tensorrt Engine

1. Download one of the following onnx models:

   • rife49_ensemble_True_scale_1_sim.onnx
   • rife48_ensemble_True_scale_1_sim.onnx
   • rife47_ensemble_True_scale_1_sim.onnx

2. Edit paths inside export_trt.py and build tensorrt engine by running:

   • python export_trt.py

3. Place the exported engine inside ComfyUI /models/tensorrt/rife directory

☀️ Usage

• Insert node by Right Click -> tensorrt -> Rife Tensorrt

🤖 Environment tested

• Ubuntu 22.04 LTS, Cuda 12.4, Tensorrt 10.4.0, Python 3.10, RTX 3070 GPU
• Windows (Not tested, but should work)

👏 Credits

• https://github.com/styler00dollar/VSGAN-tensorrt-docker
• https://github.com/Fannovel16/ComfyUI-Frame-Interpolation
• https://github.com/hzwer/ECCV2022-RIFE

License

Creative Commons Attribution-NonCommercial-ShareAlike 4.0 International (CC BY-NC-SA 4.0)