Skip to content

Commit

Permalink
Log the dtype and device used in the candle version.
Browse files Browse the repository at this point in the history
  • Loading branch information
LaurentMazare committed Sep 20, 2024
1 parent d828f0b commit 944286b
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion rust/moshi-backend/src/standalone.rs
Original file line number Diff line number Diff line change
Expand Up @@ -70,7 +70,7 @@ impl stream_both::AppStateInner {
sentencepiece::SentencePieceProcessor::open(&config.text_tokenizer_file)?;
// Warm-up code.
{
tracing::info!("warming up the model");
tracing::info!(?dtype, ?device, "warming up the model");
let mut lm_model = lm_model.clone();
let (_v, ys) = lm_model.forward(None, vec![None; config.encodec_num_codebooks])?;
let mut lp = candle_transformers::generation::LogitsProcessor::new(123, None, None);
Expand Down

0 comments on commit 944286b

Please sign in to comment.