File tree Expand file tree Collapse file tree 2 files changed +1
-3
lines changed
iotdb-core/ainode/iotdb/ainode/core Expand file tree Collapse file tree 2 files changed +1
-3
lines changed Original file line number Diff line number Diff line change @@ -209,11 +209,9 @@ def _run(
209209 inference_pipeline = load_pipeline (model_info , device = "cpu" )
210210 inputs = inference_pipeline .preprocess (inputs )
211211 if isinstance (inference_pipeline , ForecastPipeline ):
212- inputs = inference_pipeline ._preprocess (inputs )
213212 outputs = inference_pipeline .forecast (
214213 inputs , predict_length = output_length , ** inference_attrs
215214 )
216- outputs = inference_pipeline ._postprocess (outputs )
217215 elif isinstance (inference_pipeline , ClassificationPipeline ):
218216 outputs = inference_pipeline .classify (inputs )
219217 elif isinstance (inference_pipeline , ChatPipeline ):
Original file line number Diff line number Diff line change 4343
4444
4545def load_model (model_info : ModelInfo , ** model_kwargs ) -> Any :
46- if model_info .category == ModelCategory . BUILTIN or model_info . auto_map is not None :
46+ if model_info .auto_map is not None :
4747 model = load_model_from_transformers (model_info , ** model_kwargs )
4848 else :
4949 if model_info .model_type == "sktime" :
You can’t perform that action at this time.
0 commit comments