Unverified Commit 7905f363 authored by Sandro Jäckel's avatar Sandro Jäckel Committed by GitHub
Browse files

Merge pull request #331084 from GaetanLepage/optimum

parents b1fc62b0 22238b5e

pkgs/development/python-modules/optimum/default.nix

+18 −10
Original line number Diff line number Diff line
@@ -3,24 +3,30 @@ 
  buildPythonPackage,

  fetchFromGitHub,

  pythonOlder,



  # build-system

  setuptools,



  # dependencies

  coloredlogs,

  datasets,

  huggingface-hub,

  numpy,

  packaging,

  sympy,

  torch,

  transformers,



  # optional-dependencies

  diffusers,

  evaluate,

  h5py,

  huggingface-hub,

  numpy,

  onnx,

  onnxruntime,

  packaging,

  protobuf,

  setuptools,

  sympy,

  tensorflow,

  tf2onnx,

  timm,

  torch,

  transformers,

}:



buildPythonPackage rec {
@@ -39,6 +45,8 @@ buildPythonPackage rec { 


  build-system = [ setuptools ];



  pythonRelaxDeps = [ "transformers" ];



  dependencies = [

    coloredlogs

    datasets
@@ -108,12 +116,12 @@ buildPythonPackage rec { 


  pythonImportsCheck = [ "optimum" ];



  meta = with lib; {

  meta = {

    description = "Accelerate training and inference of 🤗 Transformers and 🤗 Diffusers with easy to use hardware optimization tools";

    mainProgram = "optimum-cli";

    homepage = "https://github.com/huggingface/optimum";

    changelog = "https://github.com/huggingface/optimum/releases/tag/${src.rev}";

    license = licenses.asl20;

    maintainers = with maintainers; [ natsukium ];

    license = lib.licenses.asl20;

    maintainers = with lib.maintainers; [ natsukium ];

  };

}