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

Fix allocation bug

Browse Source
This commit is contained in:
Gadersd
2023-08-05 17:05:20 -04:00
parent 2f3d94f9ad
commit 058dd1d7cc
6 changed files with 11 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
README.md
View File

@@ -19,9 +19,7 @@ export TORCH_CUDA_VERSION=cu113
```
### Step 2: Run the Sample Binary
Invoke the sample binary provided in the rust code, as shown below. Loading the burn model file
is currently very slow, but hopefully that will be rectified soon. You can also dump a torch model's weights and load that
which is currently much faster.
Invoke the sample binary provided in the rust code, as shown below:
```bash
# Arguments: <model_type(burn or dump)> <model> <unconditional_guidance_scale> <n_diffusion_steps> <prompt> <output_image>