tipi/kyutai
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fixes for the notebook.

Browse Source
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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
README.md
View File

@ -52,7 +52,7 @@ Here is how to choose which one to use:
<a href="https://huggingface.co/kyutai/stt-2.6b-en" target="_blank" style="margin: 2px;">
<img alt="Hugging Face" src="https://img.shields.io/badge/%F0%9F%A4%97%20Hugging%20Face-Model-blue" style="display: inline-block; vertical-align: middle;"/>
</a>
<a target="_blank" href="https://colab.research.google.com/github/kyutai-labs/delayed-streams-modeling/blob/main/stt_pytorch.ipynb">
<a target="_blank" href="https://colab.research.google.com/github/kyutai-labs/delayed-streams-modeling/blob/main/tts_pytorch.ipynb">
<img src="https://colab.research.google.com/assets/colab-badge.svg" alt="Open In Colab"/>
</a>

3
tts_pytorch.ipynb
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",