mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-06-30 03:37:44 +00:00
67c8befd1d
Use `parameterized` to decompose the huge test case. This will make adding ROCm support be possible. --------- Co-authored-by: Guangyun Han <guangyunhan@microsoft.com@h100vm-ort.kxelwkzfzxguje5bxvwxxs135a.gvxx.internal.cloudapp.net>
11 lines
276 B
Text
11 lines
276 B
Text
# packages used by transformers python unittest (only enabled in Linux CPU CI Pipeline)
|
|
packaging
|
|
protobuf==3.20.2
|
|
numpy==1.24.0 ; python_version < '3.12'
|
|
numpy==1.26.0 ; python_version >= '3.12'
|
|
torch
|
|
coloredlogs==15.0
|
|
transformers==4.38.0
|
|
parameterized>=0.8.1
|
|
psutil
|
|
einops
|