huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-17 02:58:50 +00:00
Code Issues Packages Projects Releases Wiki Activity

A few more tweaks.

Browse Source
This commit is contained in:
laurent
2024-03-05 22:39:06 +01:00
parent 52ed77c16f
commit b5ee026cea
2 changed files with 3 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
candle-examples/examples/metavoice/main.rs
View File

@ -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")?,
};

8
candle-transformers/src/models/metavoice.rs
View File

@ -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(