kanshan/TensorRT-LLMs
1
0
Fork 0
mirror of https://github.com/NVIDIA/TensorRT-LLM.git synced 2026-01-14 06:27:45 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

fix: Set default prompts and media for multimodal quickstart example (#3792)

Browse Source 
Signed-off-by: qixiang-99 <203170375+qixiang-99@users.noreply.github.com>
This commit is contained in:
qixiang-99 2025-04-23 22:02:28 -07:00 committed by GitHub
parent 777c40e5fa
commit 3f67a4c9d8
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
examples/pytorch/quickstart_multimodal.py
View File

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