pub fn load_b( batch_size: Option<usize>, model_file: &str, num_codebooks: Option<usize>, dev: &Device, ) -> Result<Mimi>