[AINode] Integrate PatchTST-FM-R1 as builtin forecasting model #17301
Description
Search before asking
- I searched in the issues and found nothing similar.
Motivation
Description:
Summary: This issue tracks the integration of IBM Research's PatchTST-FM-R1 into IoTDB AINode's built-in model registry, following the contribution guide documented here.
Motivation: PatchTST-FM-R1 is a SOTA patch-based Transformer foundation model for time series forecasting. Adding it expands the range of zero-shot forecasting options available to IoTDB users without requiring model training.
Implementation Plan:
Create iotdb-core/ainode/iotdb/ainode/core/model/patchtst/ directory with a PatchTSTFMPipeline extending
ForecastPipeline
Register the model in BUILTIN_HF_TRANSFORMERS_MODEL_MAP in
model_info.py
Add integration test
Reference: PR #16903 (Chronos2 integration) as the implementation pattern.
Solution
No response
Alternatives
No response
Are you willing to submit a PR?
- I'm willing to submit a PR!