51 lines
1 KiB
Nix
51 lines
1 KiB
Nix
{
|
|
lib,
|
|
buildPythonPackage,
|
|
fetchPypi,
|
|
coloredlogs,
|
|
numpy,
|
|
onnx,
|
|
packaging,
|
|
psutil,
|
|
py-cpuinfo,
|
|
py3nvml,
|
|
sympy,
|
|
}:
|
|
|
|
buildPythonPackage rec {
|
|
pname = "onnxruntime-tools";
|
|
version = "1.7.0";
|
|
format = "wheel";
|
|
|
|
# the build distribution doesn't work at all, it seems to expect the same structure
|
|
# as the github source repo.
|
|
# The github source wasn't immediately obvious how to build for this subpackage.
|
|
src = fetchPypi {
|
|
pname = "onnxruntime_tools";
|
|
inherit version;
|
|
format = "wheel";
|
|
dist = "py3";
|
|
python = "py3";
|
|
hash = "sha256-Hf+Ii1xIKsW8Yn8S4QhEX+/LPWAMQ/Y2M5dTFv5hetg=";
|
|
};
|
|
|
|
propagatedBuildInputs = [
|
|
coloredlogs
|
|
numpy
|
|
onnx
|
|
packaging
|
|
psutil
|
|
py-cpuinfo
|
|
py3nvml
|
|
sympy
|
|
];
|
|
|
|
pythonImportsCheck = [ "onnxruntime_tools" ];
|
|
|
|
meta = with lib; {
|
|
description = "Transformers Model Optimization Tool of ONNXRuntime";
|
|
homepage = "https://pypi.org/project/onnxruntime-tools/";
|
|
license = with licenses; [ mit ];
|
|
maintainers = with maintainers; [ happysalada ];
|
|
};
|
|
}
|