diff --git a/dan/decoder.py b/dan/decoder.py
index 2f63e257c687f53591509a56fc1cc0a6386b8360..3cb86963dcfbbaa930def033b5499aba2f31cb46 100644
--- a/dan/decoder.py
+++ b/dan/decoder.py
@@ -381,8 +381,6 @@ class GlobalHTADecoder(Module):
         cache=None,
         num_pred=None,
         keep_all_weights=False,
-        token_line=None,
-        token_pg=None,
     ):
         device = raw_features_1d.device