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

Musicgen text embeddings. (#726)

Browse Source 
* Musicgen text embeddings.

* Bugfix for layer norm.

* Proper position bias.

* Expose the weights.
This commit is contained in:
Laurent Mazare
2023-09-03 19:27:48 +02:00
committed by GitHub
parent bbec527bb9
commit 26cd266e65
4 changed files with 85 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
candle-examples/examples/musicgen/musicgen_model.rs
View File

@ -370,9 +370,9 @@ impl MusicgenForCausalLM {
#[derive(Debug)]
pub struct MusicgenForConditionalGeneration {
text_encoder: crate::t5_model::T5EncoderModel,
audio_encoder: crate::encodec_model::EncodecModel,
decoder: MusicgenForCausalLM,
pub text_encoder: crate::t5_model::T5EncoderModel,
pub audio_encoder: crate::encodec_model::EncodecModel,
pub decoder: MusicgenForCausalLM,
cfg: GenConfig,
}