File size: 1,246 Bytes
84669a3 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 |
from .original_generator import OriginalModelGenerator
from .f1_generator import F1ModelGenerator
from .video_generator import VideoModelGenerator
from .video_f1_generator import VideoF1ModelGenerator
from .original_with_endframe_generator import OriginalWithEndframeModelGenerator
def create_model_generator(model_type, **kwargs):
"""
Create a model generator based on the model type.
Args:
model_type: The type of model to create ("Original", "Original with Endframe", "F1", "Video", or "Video F1")
**kwargs: Additional arguments to pass to the model generator constructor
Returns:
A model generator instance
Raises:
ValueError: If the model type is not supported
"""
if model_type == "Original":
return OriginalModelGenerator(**kwargs)
elif model_type == "Original with Endframe":
return OriginalWithEndframeModelGenerator(**kwargs)
elif model_type == "F1":
return F1ModelGenerator(**kwargs)
elif model_type == "Video":
return VideoModelGenerator(**kwargs)
elif model_type == "Video F1":
return VideoF1ModelGenerator(**kwargs)
else:
raise ValueError(f"Unsupported model type: {model_type}")
|