diff options
| author | Ben Sima <ben@bsima.me> | 2024-04-01 13:30:45 -0400 |
|---|---|---|
| committer | Ben Sima <ben@bsima.me> | 2024-04-01 13:30:45 -0400 |
| commit | db373a8c727cad91d375b40a6c70b11ed73bdafb (patch) | |
| tree | a7ed24e98242b6e38f44cb0c9884718d248cc613 /Biz/Bild/Deps/exllama.nix | |
| parent | 5c8ef1bf4dff4fc7c6e66a57673a81477bcc850a (diff) | |
Add nixfmt to Lint.hs
nixfmt is the soon-to-be official formatter for Nix code, as per the NixOS
GitHub group. So I figure I should just adopt it without worrying too much about
the specifics of the formatting. I just formatted everything in one go, hence
the huge diff, oh well.
Diffstat (limited to 'Biz/Bild/Deps/exllama.nix')
| -rw-r--r-- | Biz/Bild/Deps/exllama.nix | 20 |
1 files changed, 5 insertions, 15 deletions
diff --git a/Biz/Bild/Deps/exllama.nix b/Biz/Bild/Deps/exllama.nix index 1f7e529..51a05e9 100644 --- a/Biz/Bild/Deps/exllama.nix +++ b/Biz/Bild/Deps/exllama.nix @@ -1,14 +1,6 @@ -{ lib -, sources -, buildPythonPackage -, pythonOlder +{ lib, sources, buildPythonPackage, pythonOlder , torch # tested on 2.0.1 and 2.1.0 (nightly) with cu118 -, safetensors -, sentencepiece -, ninja -, cudaPackages -, addOpenGLRunpath -, which +, safetensors, sentencepiece, ninja, cudaPackages, addOpenGLRunpath, which , gcc11 # cuda 11.7 requires g++ <12 }: @@ -36,12 +28,10 @@ buildPythonPackage rec { cudaPackages.cuda_cudart ]; - propagatedBuildInputs = [ - torch safetensors sentencepiece ninja - cudaPackages.cudatoolkit - ]; + propagatedBuildInputs = + [ torch safetensors sentencepiece ninja cudaPackages.cudatoolkit ]; - doCheck = false; # no tests currently + doCheck = false; # no tests currently pythonImportsCheck = [ "exllama" "exllama.cuda_ext" |