From 3f67a4c9d8cc42ed8b758713313370904f3b8eae Mon Sep 17 00:00:00 2001 From: qixiang-99 <203170375+qixiang-99@users.noreply.github.com> Date: Wed, 23 Apr 2025 22:02:28 -0700 Subject: [PATCH] fix: Set default prompts and media for multimodal quickstart example (#3792) Signed-off-by: qixiang-99 <203170375+qixiang-99@users.noreply.github.com> --- examples/pytorch/quickstart_multimodal.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/examples/pytorch/quickstart_multimodal.py b/examples/pytorch/quickstart_multimodal.py index ca40b532b0..4436c1a9ba 100644 --- a/examples/pytorch/quickstart_multimodal.py +++ b/examples/pytorch/quickstart_multimodal.py @@ -87,6 +87,11 @@ def parse_arguments(): def main(): args = parse_arguments() + # set prompts and media to example prompts and images if they are not provided + if args.prompt is None: + args.prompt = example_image_prompts if args.modality == "image" else example_video_prompts + if args.media is None: + args.media = example_images if args.modality == "image" else example_videos llm, sampling_params = setup_llm(args)