1 files changed, 7 insertions, 2 deletions
diff --git a/decoder/cdec.cc b/decoder/cdec.cc
index 62e28bbc..4853adfb 100644
--- a/decoder/cdec.cc
+++ b/decoder/cdec.cc
@@ -348,8 +348,13 @@ int main(int argc, char** argv) {
       // TODO check that multiple features aren't trying to set the same fid
       pffs.push_back(pff);
       late_ffs.push_back(p);
-      if (p->NumBytesContext()==0)
-        prelm_ffs.push_back(p);
+      if (has_prelm_models) {
+        int nbyte=p->NumBytesContext();
+        if (nbyte==0)
+          prelm_ffs.push_back(p);
+        else
+          cerr << "State required for feature: "<<ff<<" is "<<nbyte<<" - leaving it out of prelm pruning."<<endl;
+}
     }
   }
   ModelSet late_models(feature_weights, late_ffs);