From 81331cf535b2af338dc3c54dffe4d9c6dcce86bb Mon Sep 17 00:00:00 2001 From: Yingda Chen Date: Thu, 21 Nov 2024 20:14:31 +0800 Subject: [PATCH] fix --- modelscope/pipelines/base.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/modelscope/pipelines/base.py b/modelscope/pipelines/base.py index dc51b50a..f9e7a5a9 100644 --- a/modelscope/pipelines/base.py +++ b/modelscope/pipelines/base.py @@ -193,7 +193,7 @@ class Pipeline(ABC): # place model to cpu or gpu if (self.model or (self.has_multiple_models and self.models[0])): if not self._model_prepare: - self.prepare_model() + self.prepare_mode # simple showcase, need to support iterator type for both tensorflow and pytorch # input_dict = self._handle_input(input) @@ -207,6 +207,11 @@ class Pipeline(ABC): kwargs['preprocess_params'] = preprocess_params kwargs['forward_params'] = forward_params kwargs['postprocess_params'] = postprocess_params + if isinstance(input, list): + input = {'messages': input} + + print('call 2') + print(input) if isinstance(input, list): if batch_size is None: output = []