Explorar o código

Fix typing bug.

Kaushik Ram Sadagopan %!s(int64=2) %!d(string=hai) anos
pai
achega
5676cc9d6e
Modificáronse 1 ficheiros con 1 adicións e 1 borrados
  1. 1 1
      src/seamless_communication/models/inference/translator.py

+ 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]: