Add Zonos snippet (#1363)

Co-authored-by: Deep-Unlearning <[email protected]>
Co-authored-by: Lucain <[email protected]>
Co-authored-by: vb <[email protected]>

Author: 4 people

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

@@ -1507,4 +1507,23 @@ export const hezar = (model: ModelData): string[] => [
 
 model = Model.load("${model.id}")`,
 ];
+
+export const zonos = (model: ModelData): string[] => [
+	`# pip install git+https://github.com/Zyphra/Zonos.git
+import torchaudio
+from zonos.model import Zonos
+from zonos.conditioning import make_cond_dict
+
+model = Zonos.from_pretrained("${model.id}", device="cuda")
+
+wav, sr = torchaudio.load("speaker.wav")           # 5-10s reference clip
+speaker = model.make_speaker_embedding(wav, sr)
+
+cond  = make_cond_dict(text="Hello, world!", speaker=speaker, language="en-us")
+codes = model.generate(model.prepare_conditioning(cond))
+
+audio = model.autoencoder.decode(codes)[0].cpu()
+torchaudio.save("sample.wav", audio, model.autoencoder.sampling_rate)
+`,
+];
 //#endregion

packages/tasks/src/model-libraries.ts

@@ -1024,6 +1024,14 @@ export const MODEL_LIBRARIES_UI_ELEMENTS = {
 		countDownloads: `path_extension:"pt" OR path_extension:"safetensors"`,
 		snippets: snippets.ultralytics,
 	},
+	zonos: {
+		prettyLabel: "Zonos",
+		repoName: "Zonos",
+		repoUrl: "https://github.com/Zyphra/Zonos",
+		docsUrl: "https://github.com/Zyphra/Zonos",
+		snippets: snippets.zonos,
+		filter: false,
+	},
 	"3dtopia-xl": {
 		prettyLabel: "3DTopia-XL",
 		repoName: "3DTopia-XL",