Skip to content

Commit 12a65ad

Browse files
Update tests/benchmarks/micro_benchmarks/test_huggingface_e2e.py
Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>
1 parent 18df07b commit 12a65ad

1 file changed

Lines changed: 3 additions & 1 deletion

File tree

tests/benchmarks/micro_benchmarks/test_huggingface_e2e.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,9 @@
44
"""End-to-end integration tests for HuggingFace model loading.
55
66
These tests actually download and load models from HuggingFace Hub.
7-
Skipped when SB_TEST_CUDA=0.
7+
The test class is skipped according to ``@decorator.cuda_test``, and
8+
``test_load_model_to_gpu`` is additionally skipped when
9+
``torch.cuda.is_available()`` is false.
810
"""
911

1012
import pytest

0 commit comments

Comments
 (0)