mirror of
https://github.com/nod-ai/AMD-SHARK-Studio.git
synced 2026-04-03 03:00:17 -04:00
ORT-HF Benchmark Integration (#101)
-Add HF Benchmarker class. -Add sample to benchmark HF model. Example: ```bash python -m benchmarks.hf_model_benchmark --num_iterations=10 --model_name="microsoft/MiniLM-L12-H384-uncased" ```
This commit is contained in:
2
.github/workflows/test-models.yml
vendored
2
.github/workflows/test-models.yml
vendored
@@ -90,5 +90,5 @@ jobs:
|
||||
- name: Validate Models
|
||||
run: |
|
||||
source shark.venv/bin/activate
|
||||
pytest -k 'not benchmark'
|
||||
pytest -k 'not benchmark' --ignore=shark/tests/test_hf_benchmark.py
|
||||
|
||||
|
||||
Reference in New Issue
Block a user