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

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

@ -94,6 +94,8 @@ def generate_video(
output_fps = default(output_fps, fps_id) 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) video_model_config = config.MODEL_WEIGHT_CONFIG_LOOKUP.get(model_name, None)
if video_model_config is None: if video_model_config is None:
msg = f"Version {model_name} does not exist." msg = f"Version {model_name} does not exist."

@ -20,7 +20,7 @@ logger = logging.getLogger(__name__)
@click.option( @click.option(
"--model", "--model",
default="svd", 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( @click.option(
"--fps", default=6, type=int, help="FPS for the AI to target when generating video" "--fps", default=6, type=int, help="FPS for the AI to target when generating video"

Loading…
Cancel
Save