{ 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=";
  };
}