Chatterbox-Multilingual-TTS-API

Paused

rahul7star commited on Jan 6

Commit

909217d

verified ·

1 Parent(s): e2688ca

Update src/chatterbox/mtl_tts.py

Files changed (1) hide show

src/chatterbox/mtl_tts.py CHANGED Viewed

@@ -130,7 +130,7 @@ class Conditionals:
     @classmethod
     def load(cls, fpath, map_location="cpu"):
-        kwargs = torch.load(fpath, map_location=map_location, weights_only=True)
         return cls(T3Cond(**kwargs['t3']), kwargs['gen'])

     @classmethod
     def load(cls, fpath, map_location="cpu"):
+        kwargs = torch.load(fpath, map_location="cpu", weights_only=True)
         return cls(T3Cond(**kwargs['t3']), kwargs['gen'])