Skip to content

Commit 6a156ca

Browse files
authored
[nvbug/5376229]: Remove flash-attn dependency from test_ptp_quickstart_multimodal (#6236)
Signed-off-by: moraxu <[email protected]>
1 parent 47893cf commit 6a156ca

File tree

1 file changed

+0
-3
lines changed

1 file changed

+0
-3
lines changed

tests/integration/defs/test_e2e.py

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1911,9 +1911,6 @@ def test_ptp_quickstart_advanced_mixed_precision(llm_root, llm_venv):
19111911
])
19121912
def test_ptp_quickstart_multimodal(llm_root, llm_venv, model_name, model_path,
19131913
modality):
1914-
llm_venv.run_cmd(
1915-
['-m', 'pip', 'install', 'flash-attn==2.7.3', '--no-build-isolation'])
1916-
19171914
example_root = Path(os.path.join(llm_root, "examples", "pytorch"))
19181915
test_data_root = Path(
19191916
os.path.join(llm_models_root(), "multimodals", "test_data"))

0 commit comments

Comments
 (0)