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

[nvbug/5354825] Fix nougat test image url (#5496)

Browse Source 
Signed-off-by: Anurag Mukkara <134339030+amukkara@users.noreply.github.com>
This commit is contained in:
Anurag Mukkara 2025-06-25 19:10:18 -07:00 committed by GitHub
parent 87ead4ecbe
commit c2799d0465
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
tensorrt_llm/runtime/multimodal_model_runner.py
View File

@ -2231,6 +2231,7 @@ class MultimodalModelRunner:
filepath = hf_hub_download(
repo_id="hf-internal-testing/fixtures_docvqa",
filename="nougat_paper.png",
revision="ec57bf8c8b1653a209c13f6e9ee66b12df0fc2db",
repo_type="dataset")
images = Image.open(filepath)
elif "fuyu" in self.model_type: