alfiannajih commited on
Commit
434fca2
1 Parent(s): 5cec13a

Update g_retriever_pipeline.py

Browse files
Files changed (1) hide show
  1. g_retriever_pipeline.py +1 -1
g_retriever_pipeline.py CHANGED
@@ -57,4 +57,4 @@ class GRetrieverPipeline(Pipeline):
57
  return model_outputs
58
 
59
  def postprocess(self, model_outputs):
60
- return self.tokenizer.decode(model_outputs[0, self.input_length:])
 
57
  return model_outputs
58
 
59
  def postprocess(self, model_outputs):
60
+ return self.tokenizer.decode(model_outputs[0, self.input_length:], skip_special_tokens=True)