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

Use wgpu by default and ndarray for convert

Browse Source
This commit is contained in:
Gadersd
2023-08-08 15:32:21 -04:00
parent b87273c2be
commit 1830756917
5 changed files with 20 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -74,11 +74,11 @@ fn main() {
process::exit(1);
})
};
let sd = sd.to_device(&device);
let unconditional_context = sd.unconditional_context(&tokenizer);
let context = sd.context(&tokenizer, prompt).unsqueeze().repeat(0, 2); // generate 2 samples
let context = sd.context(&tokenizer, prompt).unsqueeze::<3>();//.repeat(0, 2); // generate 2 samples
println!("Sampling image...");
let images = sd.sample_image(context, unconditional_context, unconditional_guidance_scale, n_steps);