Add MambaSSM as a library (#802)

This PR adds https://github.com/state-spaces/mamba as an official
library on the Hub.
Let's wait for Mamba integration to be merged (coming soon) + having a
few first models uploaded on the Hub.

cc @osanseviero @NielsRogge @tridao

Related PR on Mamba side: state-spaces/mamba#471

Author: Wauplin

packages/tasks/src/model-libraries-snippets.ts

@@ -270,6 +270,12 @@ model = from_pretrained_keras("${model.id}")
 `,
 ];
 
+export const mamba_ssm = (model: ModelData): string[] => [
+	`from mamba_ssm import MambaLMHeadModel
+
+model = MambaLMHeadModel.from_pretrained("${model.id}")`,
+];
+
 export const mars5_tts = (model: ModelData): string[] => [
 	`# Install from https://github.com/Camb-ai/MARS5-TTS
 

packages/tasks/src/model-libraries.ts

@@ -298,6 +298,13 @@ export const MODEL_LIBRARIES_UI_ELEMENTS = {
 		repoName: "mindspore",
 		repoUrl: "https://github.com/mindspore-ai/mindspore",
 	},
+	"mamba-ssm": {
+		prettyLabel: "MambaSSM",
+		repoName: "MambaSSM",
+		repoUrl: "https://github.com/state-spaces/mamba",
+		filter: false,
+		snippets: snippets.mamba_ssm,
+	},
 	"mars5-tts": {
 		prettyLabel: "MARS5-TTS",
 		repoName: "MARS5-TTS",