Ben_Kosytorz/stable-diffusion-burn-vibe
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Switch to burn 0.9.0 to gain fast model io without the need for a custom recorder

Browse Source
This commit is contained in:
Gadersd
2023-08-06 19:44:13 -04:00
parent 1d1272b91f
commit c24d37df00
5 changed files with 10 additions and 97 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
Cargo.toml
View File

@@ -11,15 +11,17 @@ torch-backend = ["burn-tch"]
wgpu-backend = ["burn-wgpu"]
[dependencies.burn-tch]
version = "0.8.0"
package = "burn-tch"
git = "https://github.com/burn-rs/burn.git"
optional = true
[dependencies.burn-wgpu]
version = "0.8.0"
package = "burn-wgpu"
git = "https://github.com/burn-rs/burn.git"
optional = true
[dependencies]
burn = "0.8.0"
burn = { git = "https://github.com/burn-rs/burn.git" }
serde = {version = "1.0.171", features = ["std", "derive"]}
npy = "0.4.0"
num-traits = "0.2.15"