Spaces:

gyroing
/

Chatterbox-TTS-Persian-Farsi

Running

gyroing commited on 1 day ago

Commit

392f983

verified ·

1 Parent(s): 54a9a9f

Update src/chatterbox/mtl_tts.py

Files changed (1) hide show

src/chatterbox/mtl_tts.py CHANGED Viewed

@@ -176,7 +176,7 @@ class ChatterboxMultilingualTTS:
         s3gen = S3Gen()
         s3gen.load_state_dict(
-            torch.load(ckpt_dir / "s3gen.pt", weights_only=True)
         )
         s3gen.to(device).eval()

         s3gen = S3Gen()
         s3gen.load_state_dict(
+            torch.load(ckpt_dir / "s3gen.pt", weights_only=True,map_location=torch.device('cpu'))
         )
         s3gen.to(device).eval()