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

Addresses https://github.com/brycedrennan/imaginAIry/issues/485
pull/464/head
Bryce 4 weeks ago committed by Bryce Drennan
parent 3a9a3974ce
commit efbab3a141

@ -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