Expose AutoModelForTimeSeriesPrediction for import (#38307)

* expose AutoModelForTimeSeriesPrediction for import

* add in docs
This commit is contained in:
Jinan Zhou 2025-05-23 06:09:29 -07:00 committed by GitHub
parent a6b51e7341
commit 135163e9c5
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
4 changed files with 21 additions and 0 deletions

View File

@ -389,3 +389,9 @@ The following auto classes are available for the following multimodal tasks.
### AutoModelForImageTextToText
[[autodoc]] AutoModelForImageTextToText
## Time Series
### AutoModelForTimeSeriesPrediction
[[autodoc]] AutoModelForTimeSeriesPrediction

View File

@ -372,3 +372,10 @@ AutoModel.register(NewModelConfig, NewModel)
### AutoModelForImageTextToText
[[autodoc]] AutoModelForImageTextToText
## Time Series
### AutoModelForTimeSeriesPrediction
[[autodoc]] AutoModelForTimeSeriesPrediction

View File

@ -373,3 +373,10 @@ AutoModel.register(NewModelConfig, NewModel)
### FlaxAutoModelForVision2Seq[[transformers.FlaxAutoModelForVision2Seq]]
[[autodoc]] FlaxAutoModelForVision2Seq
## Time Series
### AutoModelForTimeSeriesPrediction[[transformers.AutoModelForTimeSeriesPrediction]]
[[autodoc]] AutoModelForTimeSeriesPrediction

View File

@ -2103,6 +2103,7 @@ __all__ = [
"AutoModelForTableQuestionAnswering",
"AutoModelForTextToSpectrogram",
"AutoModelForTextToWaveform",
"AutoModelForTimeSeriesPrediction",
"AutoModelForTokenClassification",
"AutoModelForUniversalSegmentation",
"AutoModelForVideoClassification",