modules/generators/original_with_endframe_generator.py

from .original_generator import OriginalModelGenerator

class OriginalWithEndframeModelGenerator(OriginalModelGenerator):
    """
    Model generator for the Original HunyuanVideo model with end frame support.
    This extends the Original model with the ability to guide generation toward a specified end frame.
    """
    
    def __init__(self, **kwargs):
        """
        Initialize the Original with Endframe model generator.
        """
        super().__init__(**kwargs)
        self.model_name = "Original with Endframe"
        # Inherits everything else from OriginalModelGenerator