Add files via upload

epfl-nlp · Feb 26, 2024 · 048568a · 048568a
1 parent 830e846
commit 048568a
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/BART_Rel_Pred/baseline/dataset.py b/BART_Rel_Pred/baseline/dataset.py
@@ -16,7 +16,13 @@
 
 logger = logging.getLogger(__name__)
 
-ADD_TOKENS_VALUES = ["<fact_sep>", "<rel_bos>", "<rel_eos>"]
+ADD_TOKENS_VALUES = ["<utter_sep>", "<past>", "<center>", "<future>", "<rel_bos>", "<rel_eos>",
+                     "personx", "persony", "personz", "<fact_sep>", "<eos_fact>",
+                     "<atlocation>", "<capableof>", "<causes>", "<desires>",
+                     "<hasproperty>", "<hassubevent>", "<hinderedby>", "<isafter>",
+                     "<isbefore>", "<madeupof>", "<notdesires>", "<objectuse>",
+                     "<oeffect>", "<oreact>", "<owant>", "<xattr>", "<xeffect>",
+                     "<xintent>", "<xneed>", "<xreact>", "<xreason>", "<xwant>"]
 
 
 class BaseDataset(torch.utils.data.Dataset):