[build-dependencies.anyhow]
features = ["backtrace"]
version = "1"
[build-dependencies.bindgen_cuda]
optional = true
version = "0.1.1"
[dependencies.anyhow]
features = ["backtrace"]
version = "1"
[dependencies.candle]
package = "candle-core"
version = "0.4.1"
[dependencies.candle-flash-attn]
optional = true
version = "0.4.1"
[dependencies.candle-nn]
version = "0.4.1"
[dependencies.candle-onnx]
optional = true
version = "0.4.1"
[dependencies.candle-transformers]
version = "0.4.1"
[dependencies.cudarc]
features = ["f16"]
optional = true
version = "0.10.0"
[dependencies.half]
features = ["num-traits", "use-intrinsics", "rand_distr"]
optional = true
version = "2.3.1"
[dependencies.hf-hub]
features = ["tokio"]
version = "0.3.0"
[dependencies.image]
default-features = false
features = ["jpeg", "png"]
version = "0.24.7"
[dependencies.imageproc]
default-features = false
version = "0.23.0"
[dependencies.intel-mkl-src]
features = ["mkl-static-lp64-iomp"]
optional = true
version = "0.8.1"
[dependencies.rand]
version = "0.8.5"
[dependencies.safetensors]
version = "0.4.1"
[dependencies.tokenizers]
default-features = false
features = ["onig"]
version = "0.15.0"
[features]
cuda = ["candle/cuda", "candle-nn/cuda", "candle-transformers/cuda", "dep:bindgen_cuda"]
cudnn = ["candle/cudnn"]
default = ["cuda"]
flash-attn = ["cuda", "candle-transformers/flash-attn", "dep:candle-flash-attn"]
metal = ["candle/metal", "candle-nn/metal"]
mkl = ["dep:intel-mkl-src", "candle/mkl", "candle-nn/mkl", "candle-transformers/mkl"]
nccl = ["cuda", "cudarc/nccl", "dep:half"]
onnx = ["candle-onnx"]
[package]
authors = ["Danilo Guanabara <danilo@sensorial.systems>"]
description = "Stable Diffusion"
documentation = "https://docs.rs/stable-diffusion-trainer"
edition = "2021"
homepage = "https://github.com/sensorial-systems/stable-diffusion"
keywords = ["stable-diffusion", "generative-ai", "trainer"]
license = "Apache-2.0"
name = "stable-diffusion"
readme = "README.md"
repository = "https://github.com/sensorial-systems/stable-diffusion"
version = "0.1.9"