Skip to content

Commit 164f3b5

Browse files
committed
test(anima): change e2e test GPU requirement from H100 to L4
Signed-off-by: akshatvishu <akshatnayak197@gmail.com>
1 parent 1ed3722 commit 164f3b5

2 files changed

Lines changed: 2 additions & 2 deletions

File tree

tests/e2e/offline_inference/test_anima.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -39,7 +39,7 @@ def _resolve_anima_assets() -> tuple[str, str]:
3939

4040
@pytest.mark.full_model
4141
@pytest.mark.diffusion
42-
@hardware_test(res={"cuda": "H100", "rocm": "MI325"}, num_cards=1)
42+
@hardware_test(res={"cuda": "L4", "rocm": "MI325"}, num_cards=1)
4343
@pytest.mark.parametrize("guidance_scale", [1.0, 4.0])
4444
def test_anima_text_to_image(guidance_scale):
4545
checkpoint, components = _resolve_anima_assets()

tests/e2e/online_serving/test_anima.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -39,7 +39,7 @@ def _resolve_anima_assets() -> tuple[str, str]:
3939

4040
@pytest.mark.full_model
4141
@pytest.mark.diffusion
42-
@hardware_test(res={"cuda": "H100", "rocm": "MI325"}, num_cards=1)
42+
@hardware_test(res={"cuda": "L4", "rocm": "MI325"}, num_cards=1)
4343
@pytest.mark.parametrize("guidance_scale", [1.0, 4.0])
4444
def test_anima_text_to_image(guidance_scale):
4545
checkpoint, components = _resolve_anima_assets()

0 commit comments

Comments
 (0)