Ben_Kosytorz/burn-stablediffusion-vibecode
1
0
Fork 0
mirror of https://gitea.hainer-ernst.de/rasmus/burn-stablediffusion-vibecode.git synced 2026-06-10 17:59:22 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add wgpu option

Browse Source
This commit is contained in:
Gadersd
2023-08-06 13:28:16 -04:00
parent 8a76c234e7
commit daba33ebf9
4 changed files with 49 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
Cargo.toml
View File

@@ -5,13 +5,26 @@ edition = "2021"
# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
[features]
default = ["torch-backend"]
torch-backend = ["burn-tch"]
wgpu-backend = ["burn-wgpu"]
[dependencies.burn-tch]
version = "0.8.0"
optional = true
[dependencies.burn-wgpu]
version = "0.8.0"
optional = true
[dependencies]
burn = "0.8.0"
burn-tch = "0.8.0"
serde = {version = "1.0.171", features = ["std", "derive"]}
npy = "0.4.0"
num-traits = "0.2.15"
rust_tokenizers = "8.1.0"
regex = "1.9.1"
image = "0.24.6"
bincode = {version = "2.0.0-alpha.0", features = ["std"]}
bincode = {version = "2.0.0-alpha.0", features = ["std"]}
cfg-if = "0.1"