diff options
Diffstat (limited to 'candle-examples/examples/mamba-minimal/model.rs')
| -rw-r--r-- | candle-examples/examples/mamba-minimal/model.rs | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/candle-examples/examples/mamba-minimal/model.rs b/candle-examples/examples/mamba-minimal/model.rs index 4a0a345d..7ebea76a 100644 --- a/candle-examples/examples/mamba-minimal/model.rs +++ b/candle-examples/examples/mamba-minimal/model.rs @@ -17,7 +17,7 @@ pub struct Config { impl Config { fn vocab_size(&self) -> usize { let pad = self.pad_vocab_size_multiple; - (self.vocab_size + pad - 1) / pad * pad + self.vocab_size.div_ceil(pad) * pad } fn dt_rank(&self) -> usize { |