File tree Expand file tree Collapse file tree
tests/benchmarks/micro_benchmarks Expand file tree Collapse file tree Original file line number Diff line number Diff line change 44"""End-to-end integration tests for HuggingFace model loading.
55
66These tests actually download and load models from HuggingFace Hub.
7- They are marked with @pytest.mark.integration and skipped by default
8- in CI due to network/disk requirements.
9-
10- Run with: pytest -m integration tests/benchmarks/micro_benchmarks/test_huggingface_e2e.py
7+ Skipped when SB_TEST_CUDA=0.
118"""
129
1310import pytest
1411import torch
1512
1613transformers = pytest .importorskip ('transformers' )
1714
15+ from tests .helper import decorator
1816from superbench .benchmarks .micro_benchmarks .huggingface_model_loader import HuggingFaceModelLoader
1917from superbench .benchmarks .micro_benchmarks .model_source_config import ModelSourceConfig
2018
2119
22- # Skip all tests if no GPU available (most HF models need significant resources)
23- pytestmark = pytest .mark .integration
24-
25-
20+ @decorator .cuda_test
2621class TestHuggingFaceE2E :
2722 """End-to-end tests for HuggingFace model loading."""
2823
You can’t perform that action at this time.
0 commit comments