Fix allocation bug

This commit is contained in:
Gadersd
2023-08-05 17:05:20 -04:00
committed by Ben_Kosytorz
parent a84825838d
commit ebf7c32e88
6 changed files with 11 additions and 12 deletions

View File

@@ -17,7 +17,7 @@ use burn::{
use burn_tch::{TchBackend, TchDevice};
use burn::record::{self, Recorder, FullPrecisionSettings};
use stablediffusion::binrecorder::{BinFileRecorderBuffered};
use stablediffusion::binrecorderfast::{BinFileRecorderBuffered};
fn convert_dump_to_model<B: Backend>(dump_path: &str, model_name: &str, device: &B::Device) -> Result<(), Box<dyn Error>> {
println!("Loading dump...");