fix SeqGPTPipeline import

This commit is contained in:
xingjun.wxj
2025-03-29 00:10:00 +08:00
parent c54f73c032
commit 73ac7a6666

View File

@@ -440,7 +440,7 @@ class QWenTextGenerationPipeline(Pipeline):
class SeqGPTPipeline(Pipeline):
def __init__(self, model: Union[Model, str], **kwargs):
from modelscope.utils.hf_util import AutoTokenizer
from modelscope import AutoTokenizer
if isinstance(model, str):
model_dir = snapshot_download(