-
Notifications
You must be signed in to change notification settings - Fork 55
/
Cargo.toml
50 lines (40 loc) · 1.13 KB
/
Cargo.toml
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
43
44
45
46
47
48
49
50
[package]
name = "diffusers"
version = "0.3.1"
authors = ["Laurent Mazare <[email protected]>"]
edition = "2021"
build = "build.rs"
description = "Rust implementation of the Diffusers library using Torch."
repository = "https://github.com/LaurentMazare/diffusers-rs"
keywords = ["pytorch", "deep-learning", "machine-learning", "diffusion", "transformers"]
categories = ["science"]
license = "MIT/Apache-2.0"
readme = "README.md"
exclude = [
"media/*",
]
[dependencies]
anyhow = "1"
thiserror = "1"
regex = "1.6.0"
tch = "0.13"
torch-sys = { version = "0.13", features = ["download-libtorch"] }
clap = { version = "4.0.19", optional = true, features = ["derive"] }
image = { version = "0.24.6", optional = true }
imageproc = { version = "0.23.0", optional = true }
[[example]]
name = "stable-diffusion"
required-features = ["clap"]
[[example]]
name = "stable-diffusion-img2img"
required-features = ["clap"]
[[example]]
name = "stable-diffusion-inpaint"
required-features = ["clap"]
[[example]]
name = "controlnet"
required-features = ["clap", "imageproc"]
[features]
doc-only = ["tch/doc-only"]
[package.metadata.docs.rs]
features = ["doc-only"]