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

update class ExtractiveSummarizationPipeline(Pipeline):

Browse Source
This commit is contained in:
xingjun.wang
2023-05-12 17:37:58 +08:00
parent 92270f5f23
commit 112466e069
2 changed files with 5 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
modelscope/pipelines/nlp/document_segmentation_pipeline.py
View File

@@ -6,11 +6,9 @@ from typing import Any, Dict, List, Union
import numpy as np
import torch
from datasets import Dataset
from transformers.models.bert.modeling_bert import BertConfig
from modelscope.metainfo import Pipelines
from modelscope.models import Model
from modelscope.models.nlp.ponet.configuration import PoNetConfig
from modelscope.outputs import OutputKeys
from modelscope.pipelines.base import Pipeline, Tensor
from modelscope.pipelines.builder import PIPELINES
@@ -51,11 +49,9 @@ class DocumentSegmentationPipeline(Pipeline):
auto_collate=auto_collate,
**kwargs)
kwargs = kwargs
if 'compile' in kwargs.keys():
kwargs.pop('compile')
if 'compile_options' in kwargs.keys():
kwargs.pop('compile_options')
kwargs.pop('compile', None)
kwargs.pop('compile_options', None)
self.model_dir = self.model.model_dir
self.model_cfg = self.model.model_cfg
if preprocessor is None:

7
modelscope/pipelines/nlp/extractive_summarization_pipeline.py
View File

@@ -44,11 +44,8 @@ class ExtractiveSummarizationPipeline(Pipeline):
auto_collate=auto_collate,
**kwargs)
kwargs = kwargs
if 'compile' in kwargs.keys():
kwargs.pop('compile')
if 'compile_options' in kwargs.keys():
kwargs.pop('compile_options')
kwargs.pop('compile', None)
kwargs.pop('compile_options', None)
self.model_dir = self.model.model_dir
self.model_cfg = self.model.model_cfg