From 4eac91c60bcb158c69fe5294e738198305c4f005 Mon Sep 17 00:00:00 2001 From: Fangjun Kuang Date: Mon, 8 Apr 2024 14:49:34 +0800 Subject: [PATCH] More fixes to --context-size=1 (#566) --- sherpa/csrc/online-zipformer-transducer-model.cc | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/sherpa/csrc/online-zipformer-transducer-model.cc b/sherpa/csrc/online-zipformer-transducer-model.cc index 0eb7b2d22..e564a8477 100644 --- a/sherpa/csrc/online-zipformer-transducer-model.cc +++ b/sherpa/csrc/online-zipformer-transducer-model.cc @@ -50,7 +50,9 @@ OnlineZipformerTransducerModel::OnlineZipformerTransducerModel( joiner_ = model_.attr("joiner").toModule(); context_size_ = - decoder_.attr("conv").toModule().attr("weight").toTensor().size(2); + decoder_.hasattr("conv") + ? decoder_.attr("conv").toModule().attr("weight").toTensor().size(2) + : 1; // Use 7 here since the subsampling is ((len - 7) // 2 + 1) // 2. int32_t pad_length = 7;