fix mypy

facebookresearch · Dec 11, 2024 · 2e9f21d · 2e9f21d
1 parent 5fac3ba
commit 2e9f21d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/sonar/inference_pipelines/speech.py b/sonar/inference_pipelines/speech.py
@@ -328,7 +328,7 @@ def __init__(
             tokenizer = load_sonar_tokenizer(tokenizer, progress=False)
 
         self.tokenizer = tokenizer
-        self.model = SonarEncoderDecoderModel(encoder, decoder).to(device).eval()
+        self.model = SonarEncoderDecoderModel(encoder, decoder).to(device).eval()  # type: ignore
 
         # Only quantize the model in CUDA to bypass the error "LayerNormKernelImpl" not implemented for 'Half'
         # in some CUDAs and torch versions
@@ -407,7 +407,7 @@ def __init__(
 
         if isinstance(encoder, str):
             encoder = load_sonar_speech_model(encoder, device=device, progress=False)
-        self.model = encoder.to(device).eval()
+        self.model = encoder.to(device).eval()  # type: ignore
 
         # Only quantize the model in CUDA to bypass the error "LayerNormKernelImpl" not implemented for 'Half'
         # in some CUDAs and torch versions