huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-19 11:56:45 +00:00
Code Issues Packages Projects Releases Wiki Activity

And another fix.

Browse Source
This commit is contained in:
laurent
2024-03-05 22:41:16 +01:00
parent b5ee026cea
commit 577316bc4e
1 changed files with 6 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -55,8 +55,12 @@ pub mod speaker_encoder {
layer_idx,
..Default::default()
};
let lstm =
candle_nn::lstm(cfg.mel_n_channels, cfg.model_hidden_size, c, vb_l.clone())?;
let in_c = if layer_idx == 0 {
cfg.mel_n_channels
} else {
cfg.model_hidden_size
};
let lstm = candle_nn::lstm(in_c, cfg.model_hidden_size, c, vb_l.clone())?;
lstms.push(lstm)
}
let linear = linear_b(