kleinay commited on
Commit
1ae67ef
1 Parent(s): 1c27898

Update qasrl_model_pipeline.py

Browse files
Files changed (1) hide show
  1. qasrl_model_pipeline.py +1 -1
qasrl_model_pipeline.py CHANGED
@@ -147,7 +147,7 @@ class QASRL_Pipeline(Text2TextGenerationPipeline):
147
 
148
  def postprocess(self, model_outputs):
149
  output_seq = self.tokenizer.decode(
150
- model_outputs["output_ids"][0],
151
  skip_special_tokens=False,
152
  clean_up_tokenization_spaces=False,
153
  )
 
147
 
148
  def postprocess(self, model_outputs):
149
  output_seq = self.tokenizer.decode(
150
+ model_outputs["output_ids"].squeeze(),
151
  skip_special_tokens=False,
152
  clean_up_tokenization_spaces=False,
153
  )