From d0baba3b8bffd52ed3a3c5d9d552194b9d1386ba Mon Sep 17 00:00:00 2001 From: Oussama Echakour Date: Wed, 31 Dec 2025 04:37:21 +0100 Subject: [PATCH] Fix CPU-only loading of CUDA checkpoints Add map_location to torch.load for CPU-only compatibility. --- src/chatterbox/mtl_tts.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/chatterbox/mtl_tts.py b/src/chatterbox/mtl_tts.py index 2c9cf0524..a4378b3a8 100644 --- a/src/chatterbox/mtl_tts.py +++ b/src/chatterbox/mtl_tts.py @@ -163,7 +163,7 @@ def from_local(cls, ckpt_dir, device) -> 'ChatterboxMultilingualTTS': ve = VoiceEncoder() ve.load_state_dict( - torch.load(ckpt_dir / "ve.pt", weights_only=True) + torch.load(ckpt_dir / "ve.pt", weights_only=True,map_location=torch.device(device)) ) ve.to(device).eval() @@ -176,7 +176,7 @@ def from_local(cls, ckpt_dir, device) -> 'ChatterboxMultilingualTTS': s3gen = S3Gen() s3gen.load_state_dict( - torch.load(ckpt_dir / "s3gen.pt", weights_only=True) + torch.load(ckpt_dir / "s3gen.pt", weights_only=True,map_location=torch.device(device)) ) s3gen.to(device).eval()