kanshan/TensorRT-LLMs
1
0
Fork 0
mirror of https://github.com/NVIDIA/TensorRT-LLM.git synced 2026-02-07 03:31:58 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

[nvbug/5376229]: Remove flash-attn dependency from test_ptp_quickstart_multimodal (#6236)

Browse Source 
Signed-off-by: moraxu <mguzek@nvidia.com>
This commit is contained in:
Michal Guzek 2025-07-22 13:42:20 -07:00 committed by GitHub
parent 47893cfd0c
commit 6a156caef9
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 0 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tests/integration/defs/test_e2e.py
View File

@ -1911,9 +1911,6 @@ def test_ptp_quickstart_advanced_mixed_precision(llm_root, llm_venv):
])
def test_ptp_quickstart_multimodal(llm_root, llm_venv, model_name, model_path,
modality):
llm_venv.run_cmd(
['-m', 'pip', 'install', 'flash-attn==2.7.3', '--no-build-isolation'])
example_root = Path(os.path.join(llm_root, "examples", "pytorch"))
test_data_root = Path(
os.path.join(llm_models_root(), "multimodals", "test_data"))