mirror of
https://github.com/huggingface/candle.git
synced 2025-06-16 10:38:54 +00:00
Fix the final rmsnorm for quantized-metavoice. (#2021)
This commit is contained in:
@ -235,6 +235,7 @@ pub mod transformer {
|
||||
xs = layer.forward(&xs, pos, &mask)?
|
||||
}
|
||||
xs.narrow(1, seqlen - 1, 1)?
|
||||
.contiguous()?
|
||||
.apply(&self.norm)?
|
||||
.apply(&self.output)
|
||||
}
|
||||
|
Reference in New Issue
Block a user