Ver código fonte

Fix typing bug.

Kaushik Ram Sadagopan 2 anos atrás
pai
commit
5676cc9d6e

+ 1 - 1
src/seamless_communication/models/inference/translator.py

@@ -124,7 +124,7 @@ class Translator(nn.Module):
     def synthesize_speech(
         self,
         code: List[int],
-        lang: int,
+        lang: str,
         speaker: Optional[int] = None,
         dur_prediction: Optional[bool] = True,
     ) -> Tuple[List[Tensor], int]: