Update g_retriever_pipeline.py
Browse files- 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)
|