Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
chenpangpang
transformers
Commits
720e9599
Unverified
Commit
720e9599
authored
Dec 05, 2022
by
Steven Liu
Committed by
GitHub
Dec 05, 2022
Browse files
Split autoclasses on modality (#20559)
* split autoclasses on modality * apply review * auto classes
parent
7d1c1c5b
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
132 additions
and
108 deletions
+132
-108
docs/source/en/model_doc/auto.mdx
docs/source/en/model_doc/auto.mdx
+132
-108
No files found.
docs/source/en/model_doc/auto.mdx
View file @
720e9599
...
...
@@ -74,226 +74,250 @@ Likewise, if your `NewModel` is a subclass of [`PreTrainedModel`], make sure its
[[
autodoc
]]
AutoProcessor
##
AutoModel
##
Generic
model
classes
The
following
auto
classes
are
available
for
instantiating
a
base
model
class
without
a
specific
head
.
###
AutoModel
[[
autodoc
]]
AutoModel
##
AutoModelForPreTraining
###
TFAutoModel
[[
autodoc
]]
TFAutoModel
###
FlaxAutoModel
[[
autodoc
]]
FlaxAutoModel
##
Generic
pretraining
classes
The
following
auto
classes
are
available
for
instantiating
a
model
with
a
pretraining
head
.
###
AutoModelForPreTraining
[[
autodoc
]]
AutoModelForPreTraining
##
AutoModelFor
CausalLM
##
#
TF
AutoModelFor
PreTraining
[[
autodoc
]]
AutoModelFor
CausalLM
[[
autodoc
]]
TF
AutoModelFor
PreTraining
##
AutoModelFor
DepthEstimation
##
#
Flax
AutoModelFor
PreTraining
[[
autodoc
]]
AutoModelFor
DepthEstimation
[[
autodoc
]]
Flax
AutoModelFor
PreTraining
##
AutoModelForMaskedLM
##
Natural
Language
Processing
[[
autodoc
]]
AutoModelForMaskedLM
The
following
auto
classes
are
available
for
the
following
natural
language
processing
tasks
.
##
AutoModelFor
Seq2Seq
LM
##
#
AutoModelFor
Causal
LM
[[
autodoc
]]
AutoModelFor
Seq2Seq
LM
[[
autodoc
]]
AutoModelFor
Causal
LM
##
AutoModelFor
SequenceClassification
##
#
TF
AutoModelFor
CausalLM
[[
autodoc
]]
AutoModelFor
SequenceClassification
[[
autodoc
]]
TF
AutoModelFor
CausalLM
##
AutoModelFor
MultipleChoice
##
#
Flax
AutoModelFor
CausalLM
[[
autodoc
]]
AutoModelFor
MultipleChoice
[[
autodoc
]]
Flax
AutoModelFor
CausalLM
##
AutoModelFor
NextSentencePrediction
##
#
AutoModelFor
MaskedLM
[[
autodoc
]]
AutoModelFor
NextSentencePrediction
[[
autodoc
]]
AutoModelFor
MaskedLM
##
AutoModelFor
TokenClassification
##
#
TF
AutoModelFor
MaskedLM
[[
autodoc
]]
AutoModelFor
TokenClassification
[[
autodoc
]]
TF
AutoModelFor
MaskedLM
##
AutoModelFor
QuestionAnswering
##
#
Flax
AutoModelFor
MaskedLM
[[
autodoc
]]
AutoModelFor
QuestionAnswering
[[
autodoc
]]
Flax
AutoModelFor
MaskedLM
##
AutoModelFor
TableQuestionAnswering
##
#
AutoModelFor
Seq2SeqLM
[[
autodoc
]]
AutoModelFor
TableQuestionAnswering
[[
autodoc
]]
AutoModelFor
Seq2SeqLM
##
AutoModelFor
DocumentQuestionAnswering
##
#
TF
AutoModelFor
Seq2SeqLM
[[
autodoc
]]
AutoModelFor
DocumentQuestionAnswering
[[
autodoc
]]
TF
AutoModelFor
Seq2SeqLM
##
AutoModelFor
ImageClassification
##
#
Flax
AutoModelFor
Seq2SeqLM
[[
autodoc
]]
AutoModelFor
ImageClassification
[[
autodoc
]]
Flax
AutoModelFor
Seq2SeqLM
##
AutoModelFor
Video
Classification
##
#
AutoModelFor
Sequence
Classification
[[
autodoc
]]
AutoModelFor
Video
Classification
[[
autodoc
]]
AutoModelFor
Sequence
Classification
##
AutoModelFor
Vision2Seq
##
#
TF
AutoModelFor
SequenceClassification
[[
autodoc
]]
AutoModelFor
Vision2Seq
[[
autodoc
]]
TF
AutoModelFor
SequenceClassification
##
AutoModelFor
VisualQuestionAnswering
##
#
Flax
AutoModelFor
SequenceClassification
[[
autodoc
]]
AutoModelFor
VisualQuestionAnswering
[[
autodoc
]]
Flax
AutoModelFor
SequenceClassification
##
AutoModelFor
AudioClassification
##
#
AutoModelFor
MultipleChoice
[[
autodoc
]]
AutoModelFor
AudioClassification
[[
autodoc
]]
AutoModelFor
MultipleChoice
##
AutoModelFor
AudioFrameClassification
##
#
TF
AutoModelFor
MultipleChoice
[[
autodoc
]]
AutoModelFor
AudioFrameClassification
[[
autodoc
]]
TF
AutoModelFor
MultipleChoice
##
AutoModelFor
CTC
##
#
Flax
AutoModelFor
MultipleChoice
[[
autodoc
]]
AutoModelFor
CTC
[[
autodoc
]]
Flax
AutoModelFor
MultipleChoice
##
AutoModelFor
SpeechSeq2Seq
##
#
AutoModelFor
NextSentencePrediction
[[
autodoc
]]
AutoModelFor
SpeechSeq2Seq
[[
autodoc
]]
AutoModelFor
NextSentencePrediction
##
AutoModelFor
AudioXVector
##
#
TF
AutoModelFor
NextSentencePrediction
[[
autodoc
]]
AutoModelFor
AudioXVector
[[
autodoc
]]
TF
AutoModelFor
NextSentencePrediction
##
AutoModelFor
MaskedImageModeling
##
#
Flax
AutoModelFor
NextSentencePrediction
[[
autodoc
]]
AutoModelFor
MaskedImageModeling
[[
autodoc
]]
Flax
AutoModelFor
NextSentencePrediction
##
AutoModelFor
ObjectDetec
tion
##
#
AutoModelFor
TokenClassifica
tion
[[
autodoc
]]
AutoModelFor
ObjectDetec
tion
[[
autodoc
]]
AutoModelFor
TokenClassifica
tion
##
AutoModelFor
ImageSegment
ation
##
#
TF
AutoModelFor
TokenClassific
ation
[[
autodoc
]]
AutoModelFor
ImageSegment
ation
[[
autodoc
]]
TF
AutoModelFor
TokenClassific
ation
##
AutoModelFor
SemanticSegment
ation
##
#
Flax
AutoModelFor
TokenClassific
ation
[[
autodoc
]]
AutoModelFor
SemanticSegment
ation
[[
autodoc
]]
Flax
AutoModelFor
TokenClassific
ation
##
AutoModelFor
InstanceSegmentation
##
#
AutoModelFor
QuestionAnswering
[[
autodoc
]]
AutoModelFor
InstanceSegmentation
[[
autodoc
]]
AutoModelFor
QuestionAnswering
##
AutoModelFor
ZeroShotObjectDetection
##
#
TF
AutoModelFor
QuestionAnswering
[[
autodoc
]]
AutoModelFor
ZeroShotObjectDetection
[[
autodoc
]]
TF
AutoModelFor
QuestionAnswering
##
TF
AutoModel
##
#
Flax
AutoModel
ForQuestionAnswering
[[
autodoc
]]
T
FAutoModel
[[
autodoc
]]
F
lax
AutoModel
ForQuestionAnswering
##
TFAutoModelForPreTraining
##
Computer
vision
[[
autodoc
]]
TFAutoModelForPreTraining
The
following
auto
classes
are
available
for
the
following
computer
vision
tasks
.
##
TF
AutoModelFor
CausalLM
##
#
AutoModelFor
DepthEstimation
[[
autodoc
]]
TF
AutoModelFor
CausalLM
[[
autodoc
]]
AutoModelFor
DepthEstimation
##
TFAutoModelForImageClassification
###
AutoModelForImageClassification
[[
autodoc
]]
AutoModelForImageClassification
###
TFAutoModelForImageClassification
[[
autodoc
]]
TFAutoModelForImageClassification
##
TF
AutoModelFor
SemanticSegment
ation
##
#
Flax
AutoModelFor
ImageClassific
ation
[[
autodoc
]]
T
FAutoModelFor
SemanticSegment
ation
[[
autodoc
]]
F
lax
AutoModelFor
ImageClassific
ation
##
TF
AutoModelFor
MaskedLM
##
#
AutoModelFor
VideoClassification
[[
autodoc
]]
TF
AutoModelFor
MaskedLM
[[
autodoc
]]
AutoModelFor
VideoClassification
##
TF
AutoModelFor
Seq2SeqLM
##
#
AutoModelFor
MaskedImageModeling
[[
autodoc
]]
TF
AutoModelFor
Seq2SeqLM
[[
autodoc
]]
AutoModelFor
MaskedImageModeling
##
TF
AutoModelFor
SequenceClassifica
tion
##
#
AutoModelFor
ObjectDetec
tion
[[
autodoc
]]
TF
AutoModelFor
SequenceClassifica
tion
[[
autodoc
]]
AutoModelFor
ObjectDetec
tion
##
TF
AutoModelFor
MultipleChoice
##
#
AutoModelFor
ImageSegmentation
[[
autodoc
]]
TF
AutoModelFor
MultipleChoice
[[
autodoc
]]
AutoModelFor
ImageSegmentation
##
TF
AutoModelFor
NextSentencePredic
tion
##
#
AutoModelFor
SemanticSegmenta
tion
[[
autodoc
]]
TF
AutoModelFor
NextSentencePredic
tion
[[
autodoc
]]
AutoModelFor
SemanticSegmenta
tion
##
TFAutoModelFor
TableQuestionAnswering
##
#
TFAutoModelFor
SemanticSegmentation
[[
autodoc
]]
TFAutoModelFor
TableQuestionAnswering
[[
autodoc
]]
TFAutoModelFor
SemanticSegmentation
##
TF
AutoModelFor
DocumentQuestionAnswering
##
#
AutoModelFor
InstanceSegmentation
[[
autodoc
]]
TF
AutoModelFor
DocumentQuestionAnswering
[[
autodoc
]]
AutoModelFor
InstanceSegmentation
##
TF
AutoModelFor
TokenClassifica
tion
##
#
AutoModelFor
ZeroShotObjectDetec
tion
[[
autodoc
]]
TF
AutoModelFor
TokenClassifica
tion
[[
autodoc
]]
AutoModelFor
ZeroShotObjectDetec
tion
##
TFAutoModelForQuestionAnswering
##
Audio
[[
autodoc
]]
TFAutoModelForQuestionAnswering
The
following
auto
classes
are
available
for
the
following
audio
tasks
.
##
TF
AutoModelFor
Vision2Seq
##
#
AutoModelFor
AudioClassification
[[
autodoc
]]
TF
AutoModelFor
Vision2Seq
[[
autodoc
]]
AutoModelFor
AudioClassification
##
TF
AutoModelFor
SpeechSeq2Seq
##
#
AutoModelFor
AudioFrameClassification
[[
autodoc
]]
TF
AutoModelFor
SpeechSeq2Seq
[[
autodoc
]]
AutoModelFor
AudioFrameClassification
##
Flax
AutoModel
##
#
AutoModel
ForCTC
[[
autodoc
]]
Flax
AutoModel
[[
autodoc
]]
AutoModel
ForCTC
##
Flax
AutoModelFor
CausalLM
##
#
AutoModelFor
SpeechSeq2Seq
[[
autodoc
]]
Flax
AutoModelFor
CausalLM
[[
autodoc
]]
AutoModelFor
SpeechSeq2Seq
##
Flax
AutoModelFor
PreTraining
##
#
TF
AutoModelFor
SpeechSeq2Seq
[[
autodoc
]]
F
lax
AutoModelFor
PreTraining
[[
autodoc
]]
T
FAutoModelFor
SpeechSeq2Seq
##
Flax
AutoModelFor
MaskedLM
##
#
AutoModelFor
AudioXVector
[[
autodoc
]]
Flax
AutoModelFor
MaskedLM
[[
autodoc
]]
AutoModelFor
AudioXVector
##
FlaxAutoModelForSeq2SeqLM
##
Multimodal
[[
autodoc
]]
FlaxAutoModelForSeq2SeqLM
The
following
auto
classes
are
available
for
the
following
multimodal
tasks
.
##
Flax
AutoModelFor
SequenceClassification
##
#
AutoModelFor
TableQuestionAnswering
[[
autodoc
]]
Flax
AutoModelFor
SequenceClassification
[[
autodoc
]]
AutoModelFor
TableQuestionAnswering
##
Flax
AutoModelForQuestionAnswering
##
#
TF
AutoModelFor
Table
QuestionAnswering
[[
autodoc
]]
F
lax
AutoModelForQuestionAnswering
[[
autodoc
]]
T
FAutoModelFor
Table
QuestionAnswering
##
Flax
AutoModelFor
TokenClassification
##
#
AutoModelFor
DocumentQuestionAnswering
[[
autodoc
]]
Flax
AutoModelFor
TokenClassification
[[
autodoc
]]
AutoModelFor
DocumentQuestionAnswering
##
Flax
AutoModelFor
MultipleChoice
##
#
TF
AutoModelFor
DocumentQuestionAnswering
[[
autodoc
]]
F
lax
AutoModelFor
MultipleChoice
[[
autodoc
]]
T
FAutoModelFor
DocumentQuestionAnswering
##
Flax
AutoModelFor
NextSentencePrediction
##
#
AutoModelFor
VisualQuestionAnswering
[[
autodoc
]]
Flax
AutoModelFor
NextSentencePrediction
[[
autodoc
]]
AutoModelFor
VisualQuestionAnswering
##
Flax
AutoModelFor
ImageClassification
##
#
AutoModelFor
Vision2Seq
[[
autodoc
]]
FlaxAutoModelForImageClassification
[[
autodoc
]]
AutoModelForVision2Seq
###
TFAutoModelForVision2Seq
[[
autodoc
]]
TFAutoModelForVision2Seq
##
FlaxAutoModelForVision2Seq
##
#
FlaxAutoModelForVision2Seq
[[
autodoc
]]
FlaxAutoModelForVision2Seq
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment