Skip to content

RifeTensorrt unset instance member check conflicts with real node enum #112

@mjtozaki

Description

@mjtozaki

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'

Metadata

Metadata

Assignees

No one assigned

    Labels

    bugSomething isn't workingruntime

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions