rasmus/burn-stablediffusion-vibecode
1
0
Fork 0
Code Issues Pull Requests Actions 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

2
README.md
View File

@@ -20,7 +20,7 @@ Start by downloading the SDv1-4.bin model provided on HuggingFace.
wget https://huggingface.co/Gadersd/Stable-Diffusion-Burn/resolve/main/V1/SDv1-4.bin
```
Next, set the appropriate CUDA version.
Next, set the appropriate CUDA version. It may be possible to run the model using wgpu without the need for torch in the future using `cargo run --features wgpu-backend...` but currently wgpu doesn't support buffer sizes large enough for Stable Diffusion.
```bash
export TORCH_CUDA_VERSION=cu113