akuna-embed 0.1.0

Simple text embedding models built with Burn
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72

{
  description = "akuna-embed dev shell";

  inputs = {
    nixpkgs.url = "github:NixOS/nixpkgs/nixos-unstable";
    flake-utils.url = "github:numtide/flake-utils";
  };

  outputs =
    {
      nixpkgs,
      flake-utils,
      ...
    }:
    flake-utils.lib.eachDefaultSystem (
      system:
      let
        pkgs = import nixpkgs { inherit system; };
        runtimeLibs =
          with pkgs;
          [ ]
          ++ lib.optionals stdenv.isLinux [
            libxkbcommon
            mesa
            vulkan-loader
            wayland
            libx11
            libxcursor
            libxi
            libxrandr
          ];
      in
      {
        devShells.default = pkgs.mkShell {
          packages =
            with pkgs;
            [
              curl
              gh
              git
              jq
              openssl
              pkg-config

              uv # python package manager for parity tests

              cargo # rust package manager
              cargo-deny # rust dependency license checker
              cargo-machete # rust dependency redundancy checker
              cargo-nextest # rust test runner
              clippy # rust linter
              rustc # rust compiler
              rust-analyzer # rust language server
              rustfmt # rust formatter
              sccache # rust compilation cache
            ]
            ++ runtimeLibs;

          env = {
            LIBCLANG_PATH = "${pkgs.llvmPackages.libclang.lib}/lib";
            RUSTC_WRAPPER = "${pkgs.sccache}/bin/sccache";
          }
          // pkgs.lib.optionalAttrs pkgs.stdenv.isDarwin {
            DYLD_LIBRARY_PATH = pkgs.lib.makeLibraryPath [ pkgs.zstd ];
          }
          // pkgs.lib.optionalAttrs pkgs.stdenv.isLinux {
            LD_LIBRARY_PATH = pkgs.lib.makeLibraryPath runtimeLibs;
          };
        };
      }
    );
}