fix: be forgiving of - vs _ in video decoder model name

Addresses https://github.com/brycedrennan/imaginAIry/issues/485
pull/486/head
Bryce 2 months ago
parent 1caefa51e3
commit 6474827d02

@ -94,6 +94,8 @@ def generate_video(
output_fps = default(output_fps, fps_id)
model_name = model_name.lower().replace("_", "-")
video_model_config = config.MODEL_WEIGHT_CONFIG_LOOKUP.get(model_name, None)
if video_model_config is None:
msg = f"Version {model_name} does not exist."

@ -20,7 +20,7 @@ logger = logging.getLogger(__name__)
@click.option(
"--model",
default="svd",
help="Model to use. One of: svd, svd_xt, svd_image_decoder, svd_xt_image_decoder",
help="Model to use. One of: svd, svd-xt, svd-image-decoder, svd-xt-image-decoder",
)
@click.option(
"--fps", default=6, type=int, help="FPS for the AI to target when generating video"

Loading…
Cancel
Save