rust/delayed-streams-modeling
1
0
Fork 0
mirror of https://github.com/kyutai-labs/delayed-streams-modeling.git synced 2025-12-23 03:19:57 +00:00
Code Issues Packages Projects Releases Wiki Activity

Use bfloat16 rather than half by default.

Browse Source
This commit is contained in:
Laurent
2025-07-05 23:02:58 +02:00
parent f9739881e6
commit bfc200f6ee
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tts_pytorch.ipynb
View File

@@ -55,7 +55,7 @@
"# Set everything up\n",
"checkpoint_info = CheckpointInfo.from_hf_repo(DEFAULT_DSM_TTS_REPO)\n",
"tts_model = TTSModel.from_checkpoint_info(\n",
" checkpoint_info, n_q=32, temp=0.6, device=torch.device(\"cuda\"), dtype=torch.half\n",
" checkpoint_info, n_q=32, temp=0.6, device=torch.device(\"cuda\")\n",
")\n",
"\n",
"# If you want to make a dialog, you can pass more than one turn [text_speaker_1, text_speaker_2, text_2_speaker_1, ...]\n",