{ fetchFromGitHub
, buildPythonPackage
, numpy
, packaging
, psutil
, pyyaml
, torch
}:

buildPythonPackage rec {
  name = "accelerate";
  version = "0.15.0";
  propagatedBuildInputs = [
    numpy packaging psutil pyyaml torch
  ];
  doCheck = false;
  src = fetchFromGitHub {
    owner = "huggingface";
    repo = "accelerate";
    rev = "v${version}";
    sha256 = "sha256-agfbOaa+Nm10HZkd2Y7zR3R37n+vLNsxCyxZax6O3Lo=";
  };
}