Ben_Kosytorz/burn-stablediffusion-vibecode
1
0
Fork 0
mirror of https://gitea.hainer-ernst.de/rasmus/burn-stablediffusion-vibecode.git synced 2026-06-11 02:09:21 +00:00
Code Issues 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
parent 08f7347ae5
commit 10ce5ac89d
2 changed files with 6 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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);
}