From b5ee026cea2eb41943a55c8ce02c837b13f4ccb1 Mon Sep 17 00:00:00 2001 From: laurent Date: Tue, 5 Mar 2024 22:39:06 +0100 Subject: [PATCH] A few more tweaks. --- candle-examples/examples/metavoice/main.rs | 2 +- candle-transformers/src/models/metavoice.rs | 8 ++------ 2 files changed, 3 insertions(+), 7 deletions(-) diff --git a/candle-examples/examples/metavoice/main.rs b/candle-examples/examples/metavoice/main.rs index 72b1d39a..ec725e8d 100644 --- a/candle-examples/examples/metavoice/main.rs +++ b/candle-examples/examples/metavoice/main.rs @@ -188,7 +188,7 @@ fn main() -> Result<()> { Some(w) => std::path::PathBuf::from(w), None => repo.get("first_stage.safetensors")?, }; - let second_stage_weights = match &args.first_stage_weights { + let second_stage_weights = match &args.second_stage_weights { Some(w) => std::path::PathBuf::from(w), None => repo.get("second_stage.safetensors")?, }; diff --git a/candle-transformers/src/models/metavoice.rs b/candle-transformers/src/models/metavoice.rs index 35cb30c7..2381c594 100644 --- a/candle-transformers/src/models/metavoice.rs +++ b/candle-transformers/src/models/metavoice.rs @@ -55,12 +55,8 @@ pub mod speaker_encoder { layer_idx, ..Default::default() }; - let lstm = candle_nn::lstm( - cfg.mel_n_channels, - cfg.model_hidden_size, - c, - vb_l.pp(layer_idx), - )?; + let lstm = + candle_nn::lstm(cfg.mel_n_channels, cfg.model_hidden_size, c, vb_l.clone())?; lstms.push(lstm) } let linear = linear_b(