fixing doc q/a

Aleph-Alpha · Nov 7, 2023 · ba3193d · ba3193d
1 parent f4b7509
commit ba3193d
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/atman-magma/example_document_qa_sentence_level_explain.py b/atman-magma/example_document_qa_sentence_level_explain.py
@@ -49,7 +49,7 @@
 
 document_explainer = DocumentQAExplainer(
     model,
-    explanation_delimiter = '\n',  ## for now max len = 1 token (todo: add support for longer delimiters)
+    explanation_delimiters = ['\n'],  ## for now max len = 1 token (todo: add support for longer delimiters)
     device = 'cuda:0',
     suppression_factor = 0.0,  ## suppression_factor = 0.0 -> completely erases one chunk at a time
     document = DOCUMENT
@@ -76,3 +76,4 @@
 ## save as json or print
 postprocessed_output = document_explainer.postprocess(output)
 postprocessed_output.save_as(filename = 'explanation.json')
+print('explanation.json')