diff options
| author | Nicolas Graves <ngraves@ngraves.fr> | 2024-09-09 12:08:02 +0200 |
|---|---|---|
| committer | Ricardo Wurmus <rekado@elephly.net> | 2025-04-07 16:44:27 +0200 |
| commit | 85272bac5688c7f40ad27c83fbc3db9a7da5aecc (patch) | |
| tree | 04e5de1db6455c55479438ec95be7cc2a1397f3b | |
| parent | d5c540fbd75c6667bf6f973b16eaa3947c0dc4d7 (diff) | |
gnu: Add rust-safetensors.
* gnu/packages/machine-learning.scm (rust-safetensors): New variable.
Change-Id: I2733dbae90b292ed6485666cf875d6fb899c5d12
| -rw-r--r-- | gnu/packages/machine-learning.scm | 33 |
1 files changed, 32 insertions, 1 deletions
diff --git a/gnu/packages/machine-learning.scm b/gnu/packages/machine-learning.scm index 7fdf5f37eec..8ede19722b3 100644 --- a/gnu/packages/machine-learning.scm +++ b/gnu/packages/machine-learning.scm @@ -17,7 +17,7 @@ ;;; Copyright © 2020 Edouard Klein <edk@beaver-labs.com> ;;; Copyright © 2020, 2021, 2022, 2023, 2024 Vinicius Monego <monego@posteo.net> ;;; Copyright © 2020, 2021, 2022, 2023 Maxim Cournoyer <maxim.cournoyer@gmail.com> -;;; Copyright © 2022, 2023 Nicolas Graves <ngraves@ngraves.fr> +;;; Copyright © 2022, 2023, 2024 Nicolas Graves <ngraves@ngraves.fr> ;;; Copyright © 2022 Kiran Shila <me@kiranshila.com> ;;; Copyright © 2022 Wiktor Zelazny <wzelazny@vurv.cz> ;;; Copyright © 2023 zamfofex <zamfofex@twdb.moe> @@ -53,6 +53,7 @@ #:use-module (guix utils) #:use-module (guix download) #:use-module (guix svn-download) + #:use-module (guix build-system cargo) #:use-module (guix build-system cmake) #:use-module (guix build-system copy) #:use-module (guix build-system gnu) @@ -78,6 +79,8 @@ #:use-module (gnu packages cmake) #:use-module (gnu packages cpp) #:use-module (gnu packages cran) + #:use-module (gnu packages crates-check) + #:use-module (gnu packages crates-io) #:use-module (gnu packages databases) #:use-module (gnu packages dejagnu) #:use-module (gnu packages documentation) @@ -1297,6 +1300,34 @@ and not test_wmt22_references") and reproducible BLEU, chrF, and TER scores for natural language processing.") (license license:asl2.0))) +(define-public rust-safetensors + (package + (name "rust-safetensors") + (version "0.4.3") + (source + (origin + (method url-fetch) + (uri (crate-uri "safetensors" version)) + (file-name (string-append name "-" version ".tar.gz")) + (sha256 + (base32 "1fbx56wikqcvqb4y0ym0cys68lj0v3cpanhsy5i13fkz5jr7dvcc")))) + (build-system cargo-build-system) + (arguments + `(#:cargo-inputs + (("rust-serde" ,rust-serde-1) + ("rust-serde-json" ,rust-serde-json-1)) + #:cargo-development-inputs + (("rust-criterion" ,rust-criterion-0.5) + ("rust-memmap2" ,rust-memmap2-0.9) + ("rust-proptest" ,rust-proptest-1)))) + (home-page "https://github.com/huggingface/safetensors") + (synopsis "Simple and safe way to store and distribute tensors") + (description + "This package provides a fast (zero-copy) and safe (dedicated) format for +storing tensors safely, named safetensors. They aim to be safer than their +@code{PyTorch} counterparts.") + (license license:asl2.0))) + (define-public python-sentencepiece (package (name "python-sentencepiece") |