karthikreddy-02 opened a new pull request, #17327: URL: https://github.com/apache/iotdb/pull/17327
Description This Draft PR tracks the integration of IBM's PatchTST-FM-R1 into the AINode built-in model registry, resolving [Issue #17301](https://github.com/apache/iotdb/issues/17301). Current Progress: Imported custom patchtst_fm architecture files ( configuration_patchtst_fm.py and modeling_patchtst_fm.py ) from the IBM Granite TSFM repository, as they are not natively supported in standard transformers yet. Registered the model in BUILTIN_HF_TRANSFORMERS_MODEL_MAP under model_info.py . Next Steps (WIP): Implement PatchTSTFMPipeline extending ForecastPipeline for tensor reshaping. Add integration tests. This PR has: x been self-reviewed. added documentation for new or modified features or behaviors. added Javadocs for most classes and all non-trivial methods. added or updated version, license, or notice information added comments explaining the "why" and the intent of the code wherever would not be obvious for an unfamiliar reader. added unit tests or modified existing tests to cover new code paths, ensuring the threshold for code coverage. added integration tests. been tested in a test IoTDB cluster. Key changed/added classes (or packages if there are too many classes) in this PR iotdb.ainode.core.model.patchtst_fm.configuration_patchtst_fm (Added) iotdb.ainode.core.model.patchtst_fm.modeling_patchtst_fm (Added) iotdb.ainode.core.model.model_info.BUILTIN_HF_TRANSFORMERS_MODEL_MAP (Modified) -- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. To unsubscribe, e-mail: [email protected] For queries about this service, please contact Infrastructure at: [email protected]
