More fixes to --context-size=1 (#566)

k2-fsa · Apr 8, 2024 · 4eac91c · 4eac91c
1 parent 98f9429
commit 4eac91c
Showing 1 changed file with 3 additions and 1 deletion.
diff --git 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;