From e662431acf30e167e2d63af01090b015f89b6e3b Mon Sep 17 00:00:00 2001
From: Laurent Mazare <laurent.mazare@gmail.com>
Date: Sat, 6 Apr 2024 19:35:01 +0200
Subject: [PATCH] Fix the final rmsnorm for quantized-metavoice. (#2021)

---
 candle-transformers/src/models/quantized_metavoice.rs | 1 +
 1 file changed, 1 insertion(+)

diff --git a/candle-transformers/src/models/quantized_metavoice.rs b/candle-transformers/src/models/quantized_metavoice.rs
index 84c0388c..947ab750 100644
--- a/candle-transformers/src/models/quantized_metavoice.rs
+++ b/candle-transformers/src/models/quantized_metavoice.rs
@@ -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)
         }