Mirrors/modelscope
1
0
Fork 0
mirror of https://github.com/modelscope/modelscope.git synced 2025-12-25 04:29:22 +01:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #289 from Restry/patch-1

Browse Source 
修复 fid_dialogue_pipeline.py 在cpu下推理时的异常
This commit is contained in:
Xingjun.Wang
2023-05-12 15:38:53 +08:00
committed by GitHub
parent 75410651ba 1f2bb82ea6
commit 72b617dab9
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
modelscope/pipelines/nlp/fid_dialogue_pipeline.py
View File

@@ -191,8 +191,8 @@ class FidDialoguePipeline(Pipeline):
def postprocess(self, inputs: TokenGeneratorOutput,
**postprocess_params) -> Dict[str, Any]:
if torch.cuda.is_available():
hypotheses = inputs.sequences.detach().cpu().tolist()
# if torch.cuda.is_available():
hypotheses = inputs.sequences.detach().cpu().tolist()
response = self.preprocessor_tokenizer.decode(
hypotheses[0], skip_special_tokens=self.is_t5)