Fix: Change embeddings to embedding (#2738)

fix: change embeddings to embedding
Co-authored-by: Ubuntu <ubuntu@ip-172-31-28-135.us-west-2.compute.internal>

server/text_generation_server/models/custom_modeling/mamba_modeling.py

@@ -212,7 +212,7 @@ class MambaModel(nn.Module):
         try:
             self.lm_head = SpeculativeHead.load(config, f"{prefix}.embeddings", weights)
         except RuntimeError:
-            self.lm_head = SpeculativeHead.load(config, f"{prefix}.embeddings", weights)
+            self.lm_head = SpeculativeHead.load(config, f"{prefix}.embedding", weights)
         self.config = config
 
     def forward(