Skip to content

Commit e4ae9cd

Browse files
committed
revert updates
1 parent 92894c6 commit e4ae9cd

File tree

2 files changed

+1
-3
lines changed

2 files changed

+1
-3
lines changed

iotdb-core/ainode/iotdb/ainode/core/manager/inference_manager.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff 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):

iotdb-core/ainode/iotdb/ainode/core/model/model_loader.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -43,7 +43,7 @@
4343

4444

4545
def 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":

0 commit comments

Comments
 (0)