[ASR] Fix type error in jasper (#7636) (#7653)

Signed-off-by: Ryan <rlangman@nvidia.com> Co-authored-by: Ryan Langman <rlangman@nvidia.com>
NVIDIA · Oct 6, 2023 · 5b6209d · 5b6209d
1 parent d08d1c5
commit 5b6209d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/nemo/collections/asr/parts/submodules/jasper.py b/nemo/collections/asr/parts/submodules/jasper.py
@@ -375,7 +375,7 @@ def update_masked_length(self, max_len, seq_range=None, device=None):
             self.lens = self.lens.to(device)
         else:
             self.lens = seq_range
-            self.max_len = max_len
+            self.max_len = torch.tensor(max_len)
 
     def mask_input(self, x, lens):
         max_len = x.size(2)