diffusers-burn/Cargo.toml at f40d24d98ca802232703c2a0bb722fd004b0d41f · oxiglade/diffusers-burn · GitHub

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
[package]
name = "diffusers-burn"
version = "0.1.0"
edition = "2021"

[features]
default = ["burn/default", "std"]
std = ["burn/std", "regex", "thiserror", "burn-store", "safetensors", "memmap2"]

# Backend
accelerate = ["burn/accelerate"]
cuda = ["burn/cuda"]

ndarray = ["burn/ndarray"]
torch = ["burn/tch"]
wgpu = ["burn/wgpu"]

[dependencies]
burn = { version = "0.20.1", default-features = false }
burn-store = { version = "0.20.1", optional = true, features = ["safetensors"] }
num-traits = { version = "0.2.18", default-features = false }
regex = { version = "1.10", optional = true }
safetensors = { version = "0.4", optional = true }
memmap2 = { version = "0.9", optional = true }
serde = { version = "1.0.197", default-features = false, features = [
    "derive",
    "alloc",
] }
thiserror = { version = "1.0", optional = true }

[[example]]
name = "stable-diffusion"
required-features = ["std"]

[dev-dependencies]
anyhow = "1.0"
clap = { version = "4.4", features = ["derive", "env"] }
dirs = "5.0"
flate2 = "1.0"
hf-hub = "0.3"
image = "0.25"
ureq = "2.9"