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 + 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",