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

Switch to burn 0.9.0 to gain fast model io without the need for a custom recorder

Browse Source
This commit is contained in:
Gadersd
2023-08-06 19:44:13 -04:00
committed by Ben_Kosytorz
parent 1b5426c887
commit 68d4310aa8
5 changed files with 10 additions and 97 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
src/bin/convert/main.rs
View File

@@ -22,8 +22,7 @@ cfg_if::cfg_if! {
}
}
use burn::record::{self, Recorder, FullPrecisionSettings};
use stablediffusion::binrecorderfast::{BinFileRecorderBuffered};
use burn::record::{self, Recorder, BinFileRecorder, FullPrecisionSettings};
fn convert_dump_to_model<B: Backend>(dump_path: &str, model_name: &str, device: &B::Device) -> Result<(), Box<dyn Error>> {
println!("Loading dump...");
@@ -36,7 +35,7 @@ fn convert_dump_to_model<B: Backend>(dump_path: &str, model_name: &str, device:
}
fn save_model_file<B: Backend>(model: StableDiffusion<B>, name: &str) -> Result<(), record::RecorderError> {
BinFileRecorderBuffered::<FullPrecisionSettings>::new()
BinFileRecorder::<FullPrecisionSettings>::new()
.record(
model.into_record(),
name.into(),