From 835e6fc41c8aa6a4627161252b05443afed4c9dd Mon Sep 17 00:00:00 2001 From: Jinan Zhou Date: Thu, 22 May 2025 15:14:20 -0700 Subject: [PATCH 1/2] expose AutoModelForTimeSeriesPrediction for import --- src/transformers/models/auto/modeling_auto.py | 1 + 1 file changed, 1 insertion(+) diff --git a/src/transformers/models/auto/modeling_auto.py b/src/transformers/models/auto/modeling_auto.py index 5fe74444cb20..aa88c6a4241e 100644 --- a/src/transformers/models/auto/modeling_auto.py +++ b/src/transformers/models/auto/modeling_auto.py @@ -2103,6 +2103,7 @@ def from_pretrained(cls, pretrained_model_name_or_path, *model_args, **kwargs): "AutoModelForTableQuestionAnswering", "AutoModelForTextToSpectrogram", "AutoModelForTextToWaveform", + "AutoModelForTimeSeriesPrediction", "AutoModelForTokenClassification", "AutoModelForUniversalSegmentation", "AutoModelForVideoClassification", From 96e1c3576c643b1f387d014fc2809dbdc361bdde Mon Sep 17 00:00:00 2001 From: Jinan Zhou Date: Thu, 22 May 2025 15:38:47 -0700 Subject: [PATCH 2/2] add in docs --- docs/source/en/model_doc/auto.md | 6 ++++++ docs/source/ja/model_doc/auto.md | 7 +++++++ docs/source/ko/model_doc/auto.md | 7 +++++++ 3 files changed, 20 insertions(+) diff --git a/docs/source/en/model_doc/auto.md b/docs/source/en/model_doc/auto.md index afe343228f25..adab8591e293 100644 --- a/docs/source/en/model_doc/auto.md +++ b/docs/source/en/model_doc/auto.md @@ -389,3 +389,9 @@ The following auto classes are available for the following multimodal tasks. ### AutoModelForImageTextToText [[autodoc]] AutoModelForImageTextToText + +## Time Series + +### AutoModelForTimeSeriesPrediction + +[[autodoc]] AutoModelForTimeSeriesPrediction diff --git a/docs/source/ja/model_doc/auto.md b/docs/source/ja/model_doc/auto.md index 492c46c79ea9..27030a264f57 100644 --- a/docs/source/ja/model_doc/auto.md +++ b/docs/source/ja/model_doc/auto.md @@ -372,3 +372,10 @@ AutoModel.register(NewModelConfig, NewModel) ### AutoModelForImageTextToText [[autodoc]] AutoModelForImageTextToText + +## Time Series + +### AutoModelForTimeSeriesPrediction + +[[autodoc]] AutoModelForTimeSeriesPrediction + diff --git a/docs/source/ko/model_doc/auto.md b/docs/source/ko/model_doc/auto.md index cda00adc33a6..45c2f917a42c 100644 --- a/docs/source/ko/model_doc/auto.md +++ b/docs/source/ko/model_doc/auto.md @@ -373,3 +373,10 @@ AutoModel.register(NewModelConfig, NewModel) ### FlaxAutoModelForVision2Seq[[transformers.FlaxAutoModelForVision2Seq]] [[autodoc]] FlaxAutoModelForVision2Seq + +## Time Series + +### AutoModelForTimeSeriesPrediction[[transformers.AutoModelForTimeSeriesPrediction]] + +[[autodoc]] AutoModelForTimeSeriesPrediction +