diff options
Diffstat (limited to 'candle-examples/examples/musicgen/musicgen_model.rs')
| -rw-r--r-- | candle-examples/examples/musicgen/musicgen_model.rs | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/candle-examples/examples/musicgen/musicgen_model.rs b/candle-examples/examples/musicgen/musicgen_model.rs index 751e0226..7e272fd7 100644 --- a/candle-examples/examples/musicgen/musicgen_model.rs +++ b/candle-examples/examples/musicgen/musicgen_model.rs @@ -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, } |