candle_embed 0.1.2

A simple, CUDA or CPU powered, library for creating vector embeddings using Candle and models from Hugging Face
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

[dependencies.anyhow]
version = "1.0.82"

[dependencies.candle]
default-features = false
package = "candle-core"
version = "0.4.1"

[dependencies.candle-transformers]
version = "0.4.1"

[dependencies.candle_nn]
default-features = false
package = "candle-nn"
version = "0.4.1"

[dependencies.hf-hub]
features = ["tokio"]
version = "0.3.2"

[dependencies.serde_json]
version = "1.0.116"

[dependencies.tokenizers]
version = "0.19.1"

[features]
cuda = ["candle/cuda", "candle/cudnn", "candle_nn/cuda"]
default = ["cuda"]

[package]
categories = ["cuda", "embeddings", "huggingface", "search", "vectors"]
description = "A simple, CUDA or CPU powered, library for creating vector embeddings using Candle and models from Hugging Face"
edition = "2021"
homepage = "https://github.com/shelbyJenkins/candle_embed"
keywords = ["cuda", "embeddings", "huggingface", "search", "vectors"]
license = "MIT"
name = "candle_embed"
readme = "README.md"
repository = "https://github.com/shelbyJenkins/candle_embed"
version = "0.1.2"