Files
candle/candle-transformers/Cargo.toml
Nicolas Patry 4ed56d7861 Removing cuda default.
Seems very important for a lot of exploring users usually on laptop
without GPUs.

Adding more README instructions in a follow up.
2023-07-14 16:52:15 +02:00

26 lines
833 B
TOML

[package]
name = "candle-transformers"
version = "0.1.0"
edition = "2021"
description = "Pretrained models and inference API for the candle ML framework."
repository = "https://github.com/LaurentMazare/candle"
keywords = ["blas", "tensor", "machine-learning"]
categories = ["science"]
license = "MIT/Apache-2.0"
readme = "README.md"
[dependencies]
candle = { path = "../candle-core", default-features=false }
candle-hub = { path = "../candle-hub" }
candle-nn = { path = "../candle-nn", default-features=false }
intel-mkl-src = {version="0.8.1", optional=true, features = ["mkl-dynamic-lp64-iomp"]}
tokenizers = { version = "0.13.3", default-features=false, features=["onig"] }
rand = "0.8.5"
wav = "1.0.0"
[features]
default = []
cuda = ["candle/cuda", "candle-nn/cuda"]
mkl = ["dep:intel-mkl-src", "candle/mkl", "candle-nn/mkl"]