Dispatch data in the bfloat16 format and let downstream deal with

the conversion. This fixes things on Metal.
sippy · May 27, 2024 · 252a7ca · 252a7ca
1 parent 86447f6
commit 252a7ca
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/HelloSippyTTSRT/HelloSippyRTPipe.py b/HelloSippyTTSRT/HelloSippyRTPipe.py
@@ -249,7 +249,7 @@ def unbatch_and_dispatch(self, state:HelloSippyPipeStateBatched):
                 endoff = min(asize, asize - (((state.idx - ends_at) * stepsize) if (ends_at:=state.ends_at[i].item()) >=0 else 0))
                 assert startoff <= endoff
                 if startoff != endoff:
-                    dispatch(audio[i][startoff:endoff].to(torch.float16).cpu())
+                    dispatch(audio[i][startoff:endoff].cpu())
                 if ends_at >= 0 and ends_at <= end_idx:
                     dispatch(None)
                     state.dispatch[i] = None