diff options
| author | Ben Sima <ben@bsima.me> | 2022-12-28 19:53:55 -0500 |
|---|---|---|
| committer | Ben Sima <ben@bsima.me> | 2022-12-28 19:53:55 -0500 |
| commit | c3b955145998d39df39370671585a271ca6f80f0 (patch) | |
| tree | 33614e03c966d205e2eadaf4dd183f52618afebc /Biz/Bild/Deps/accelerate.nix | |
| parent | 11e480c4b13808f12bc3f5db2765cebebf1aaf46 (diff) | |
Get ava GPT chatbot prototype working
Mostly thid required packaging up some deps, but also had to recompile
stuff with cuda support.
Diffstat (limited to 'Biz/Bild/Deps/accelerate.nix')
| -rw-r--r-- | Biz/Bild/Deps/accelerate.nix | 24 |
1 files changed, 24 insertions, 0 deletions
diff --git a/Biz/Bild/Deps/accelerate.nix b/Biz/Bild/Deps/accelerate.nix new file mode 100644 index 0000000..5d00f14 --- /dev/null +++ b/Biz/Bild/Deps/accelerate.nix @@ -0,0 +1,24 @@ +{ 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="; + }; +} + |