mirror of
https://github.com/storytold/artcraft.git
synced 2026-11-01 16:08:30 +00:00
84 lines
2.4 KiB
TOML
84 lines
2.4 KiB
TOML
[package]
|
|
name = "ml_models"
|
|
edition = "2021"
|
|
version = "0.0.1"
|
|
authors = ["storyteller"]
|
|
publish = false
|
|
description = ""
|
|
license = ""
|
|
repository = ""
|
|
|
|
[lib]
|
|
name = "ml_models"
|
|
path = "src/lib.rs"
|
|
crate-type = ["staticlib", "cdylib", "rlib"]
|
|
|
|
[package.metadata.cargo-udeps.ignore]
|
|
normal = ["workspace-build-acceleration"]
|
|
|
|
[dependencies]
|
|
|
|
# Workspace hack (faster compile times)
|
|
workspace-build-acceleration = { workspace = true }
|
|
|
|
# Internal
|
|
filesys = { path = "../../../files/filesys" }
|
|
image_algorithms = { path = "../image_algorithms" }
|
|
memory_caching = { path = "../../../caching/memory_caching" }
|
|
ml_weights_registry = { path = "../ml_weights_registry" }
|
|
primitives = { path = "../../../data/primitives" }
|
|
|
|
# Bypass cudarc issues on Windows https://github.com/coreylowman/cudarc/issues/332
|
|
cudarc = "0.13.7"
|
|
|
|
# External
|
|
anyhow = { workspace = true }
|
|
base64 = { workspace = true }
|
|
bytes = "1.10"
|
|
candle-core = { path = "../../../../vendor/candle/candle-core", features = ["cuda"] }
|
|
candle-nn = { path = "../../../../vendor/candle/candle-nn", features = ["cuda"] }
|
|
candle-transformers = { path = "../../../../vendor/candle/candle-transformers", features = ["cuda"] }
|
|
const_format = "0.2.34"
|
|
directories = "6.0.0"
|
|
dirs = "6.0.0"
|
|
futures = "0.3"
|
|
half = { version = "2.3.1", features = ["num-traits", "use-intrinsics", "rand_distr"] }
|
|
hf-hub = "0.3.2"
|
|
image = "0.25.5"
|
|
log = "0.4"
|
|
ndarray = "0.15.6"
|
|
nvidia-checker = "0.1.2"
|
|
once_cell = "1.20.3"
|
|
ort = { version = "1.16.2", features = ["cuda"] }
|
|
rand = "0.8.5"
|
|
# NB(bt,2025-03-10): default-features = false on Windows while I debug openssl / rustls
|
|
reqwest = { version = "0.12.12", features = ["stream"], default-features = false }
|
|
serde = { version = "1.0", features = ["derive"] }
|
|
serde_json = "1.0"
|
|
serde_yaml = "0.9"
|
|
tauri = { version = "2.2.4", features = [] }
|
|
tauri-plugin-log = "2"
|
|
tempdir = "0.3.7"
|
|
tempfile = "3.18.0"
|
|
tokenizers = "0.15.2"
|
|
tokio = { version = "1.42", features = ["rt", "rt-multi-thread", "fs"] }
|
|
tokio-util.workspace = true
|
|
|
|
# tracing-subscriber = { version = "0.3", features = [ "env-filter", "fmt" ] }
|
|
|
|
[features]
|
|
default = ["cuda"]
|
|
cuda = [
|
|
"candle-core/cuda",
|
|
"candle-nn/cuda",
|
|
"candle-transformers/cuda",
|
|
# NB(bt,2025-03-10): Building without Flash attention on Windows for now to bypass linker issues
|
|
"candle-transformers/flash-attn",
|
|
"candle-core/cudnn"
|
|
]
|
|
|
|
[dev-dependencies]
|
|
libloading = "0.8.6"
|
|
serial_test = "3.2.0"
|
|
assertor.workspace = true
|