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

Correct a few trivial bugs

Browse Source
This commit is contained in:
Gadersd
2023-08-05 13:46:14 -04:00
committed by Ben_Kosytorz
parent 6f9f4f1f61
commit bb80c045e0
2 changed files with 6 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
README.md
View File

@@ -19,11 +19,13 @@ export TORCH_CUDA_VERSION=cu113
```
### Step 2: Run the Sample Binary
Invoke the sample binary provided in the rust code, as shown below:
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.
```bash
# Arguments: model unconditional_guidance_scale n_diffusion_steps prompt output_image
cargo run --release --bin sample SDv1-4 7.5 20 "A half-eaten apple sitting on a desk." apple.png
# Arguments: <model_type(burn or dump)> <model> <unconditional_guidance_scale> <n_diffusion_steps> <prompt> <output_image>
cargo run --release --bin sample burn SDv1-4 7.5 20 "A half-eaten apple sitting on a desk." apple
```
This command will generate an image according to the provided prompt, which will be saved as 'apple.png'.

2
src/bin/sample/main.rs
View File

@@ -29,7 +29,7 @@ fn main() {
let device = TchDevice::Cuda(0);
let args: Vec<String> = std::env::args().collect();
if args.len() != 6 {
if args.len() != 7 {
eprintln!("Usage: {} <model_type(burn or dump)> <model_name> <unconditional_guidance_scale> <n_diffusion_steps> <prompt> <output_image_name>", args[0]);
process::exit(1);
}