mirror of
https://github.com/brycedrennan/imaginAIry
synced 2024-10-31 03:20:40 +00:00
fix: be forgiving of - vs _ in video decoder model name
Addresses https://github.com/brycedrennan/imaginAIry/issues/485
This commit is contained in:
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…
Reference in New Issue
Block a user