Fixes for the notebook.

This commit is contained in:
laurent
2025-07-03 13:05:00 +02:00
parent 1cd9529f65
commit 25574aa104
2 changed files with 3 additions and 2 deletions

View File

@@ -7,7 +7,7 @@
"metadata": {},
"outputs": [],
"source": [
"!pip install git+https://git@github.com/kyutai-labs/moshi#egg=moshi&subdirectory=moshi"
"!pip install \"git+https://git@github.com/kyutai-labs/moshi#egg=moshi&subdirectory=moshi\""
]
},
{
@@ -54,6 +54,7 @@
"tts_model = TTSModel.from_checkpoint_info(\n",
" checkpoint_info, n_q=32, temp=0.6, device=torch.device(\"cuda\"), dtype=torch.half\n",
")\n",
"tts_model.mimi.streaming_forever(1)\n",
"\n",
"# You could also generate multiple audios at once by passing a list of texts.\n",
"entries = tts_model.prepare_script([text], padding_between=1)\n",