-
Notifications
You must be signed in to change notification settings - Fork 40
Closed
Labels
Description
Custom node: https://github.com/yuvraj108c/ComfyUI-Rife-Tensorrt , head as of 2025/09/20
Node RifeTensorrt does a check to see if an instance member was set:
if (not hasattr(self, 'engine') or self.engine_label != engine):
The real node (did not confirm other types of nodes) generated for RifeTensorrt has a nested enum that conflicts with that check:
class RifeTensorrt:
'''```
def RifeTensorrt(
frames: Image,
engine: RifeTensorrt.engine | str = 'rife49_ensemble_True_scale_1_sim.engine',
clear_cache_after_n_frames: int = 100,
multiplier: int = 2,
use_cuda_graph: bool = True,
keep_model_loaded: bool = False
) -> Image
```
⚡ Rife Tensorrt
Module: `custom_nodes/ComfyUI-Rife-Tensorrt`
Category: tensorrt
- `clear_cache_after_n_frames`: 1 ≤ `clear_cache_after_n_frames` ≤ 1000
- `multiplier`: 1 ≤ `multiplier`
Output node.
'''
def __new__(cls, frames: Image, engine: RifeTensorrt.engine | str = 'rife49_ensemble_True_scale_1_sim.engine', clear_cache_after_n_frames: int = 100, multiplier: int = 2, use_cuda_graph: bool = True, keep_model_loaded: bool = False) -> Image: ...
class engine(StrEnum):
rife49_ensemble_True_scale_1_sim_engine = 'rife49_ensemble_True_scale_1_sim.engine'
Result is an error raised when it gets farther than it should: AttributeError: 'RifeTensorrt' object has no attribute 'engine_label'
Reactions are currently unavailable