flake/packages/accelerate/default.nix
Max 079f83ba45 Major Upgrade/Refactor
- KoboldAI is no longer upstream maintained, so is now deprecated in
  nixified.ai

- InvokeAI: v2.3.1.post2 -> v3.3.0post3
- textgen: init at v1.7
- treewide: update flake inputs including nixpkgs
- treewide: add a bunch of new dependencies and upgrade old ones
2023-10-19 22:13:37 +01:00

26 lines
736 B
Nix

# WARNING: This file was automatically generated. You should avoid editing it.
# If you run pynixify again, the file will be either overwritten or
# deleted, and you will lose the changes you made to it.
{ buildPythonPackage, fetchPypi, lib, numpy, packaging, psutil, pyyaml, torch, huggingface-hub }:
buildPythonPackage rec {
pname = "accelerate";
version = "0.23.0";
src = fetchPypi {
inherit pname version;
sha256 = "sha256-ITnSGfqaN3c8QnnJr+vp9oHy8p6FopsL6NdiV72OSr4=";
};
propagatedBuildInputs = [ numpy packaging psutil pyyaml torch huggingface-hub ];
# TODO FIXME
doCheck = false;
meta = with lib; {
description = "Accelerate";
homepage = "https://github.com/huggingface/accelerate";
};
}