From 75a2c667cb02ed835d2f6e7cd1112900bdd55780 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?M=C3=A9lodie=20Boillet?= <boillet@teklia.com>
Date: Thu, 8 Jun 2023 15:44:50 +0000
Subject: [PATCH] Remove token_line and token_pg from GlobalHTADecoder

---
 dan/decoder.py | 2 --
 1 file changed, 2 deletions(-)

diff --git a/dan/decoder.py b/dan/decoder.py
index 2f63e257..3cb86963 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
 
-- 
GitLab