JymNils commited on
Commit
a930ed8
·
verified ·
1 Parent(s): f27978a

Update src/chatterbox/mtl_tts.py

Browse files
Files changed (1) hide show
  1. src/chatterbox/mtl_tts.py +1 -1
src/chatterbox/mtl_tts.py CHANGED
@@ -176,7 +176,7 @@ class ChatterboxMultilingualTTS:
176
 
177
  s3gen = S3Gen()
178
  s3gen.load_state_dict(
179
- torch.load(ckpt_dir / "s3gen.pt", weights_only=True, map_location=cpu)
180
  )
181
  s3gen.to(device).eval()
182
 
 
176
 
177
  s3gen = S3Gen()
178
  s3gen.load_state_dict(
179
+ torch.load(ckpt_dir / "s3gen.pt", weights_only=True, map_location='cpu')
180
  )
181
  s3gen.to(device).eval()
182