diff --git a/docs/source/inference.mdx b/docs/source/inference.mdx index e0b60baa2e..9e68267079 100644 --- a/docs/source/inference.mdx +++ b/docs/source/inference.mdx @@ -28,8 +28,12 @@ As shown in the table below, each task is associated with a class enabling to au | `image-classification` | `OVModelForImageClassification` | | `feature-extraction` | `OVModelForFeatureExtraction` | | `fill-mask` | `OVModelForMaskedLM` | +| `image-classification` | `OVModelForImageClassification` | +| `audio-classification` | `OVModelForAudioClassification` | | `text-generation` | `OVModelForCausalLM` | | `text2text-generation` | `OVModelForSeq2SeqLM` | +| `automatic-speech-recognition` | `OVModelForSpeechSeq2Seq` | +| `image-to-text` | `OVModelForVision2Seq` | ### Export diff --git a/docs/source/reference_ov.mdx b/docs/source/reference_ov.mdx index 4c5ede653e..32385eae00 100644 --- a/docs/source/reference_ov.mdx +++ b/docs/source/reference_ov.mdx @@ -14,56 +14,113 @@ See the License for the specific language governing permissions and limitations under the License. --> -# Reference +# Models -## OVModelForFeatureExtraction +## Natural Language Processing -[[autodoc]] openvino.modeling.OVModelForFeatureExtraction +The following classes are available for the following natural language processing tasks. + +### OVModelForCausalLM + +[[autodoc]] openvino.modeling_decoder.OVModelForCausalLM + - forward + - generate -## OVModelForMaskedLM +### OVModelForMaskedLM [[autodoc]] openvino.modeling.OVModelForMaskedLM + - forward + +### OVModelForSeq2SeqLM + +[[autodoc]] openvino.modeling_seq2seq.OVModelForSeq2SeqLM + - forward -## OVModelForQuestionAnswering +### OVModelForQuestionAnswering [[autodoc]] openvino.modeling.OVModelForQuestionAnswering + - forward -## OVModelForSequenceClassification +### OVModelForSequenceClassification [[autodoc]] openvino.modeling.OVModelForSequenceClassification + - forward -## OVModelForTokenClassification +### OVModelForTokenClassification [[autodoc]] openvino.modeling.OVModelForTokenClassification + - forward -## OVModelForAudioClassification + +## Audio + +The following classes are available for the following audio tasks. + +### OVModelForAudioClassification [[autodoc]] openvino.modeling.OVModelForAudioClassification + - forward -## OVModelForAudioFrameClassification +### OVModelForAudioFrameClassification [[autodoc]] openvino.modeling.OVModelForAudioFrameClassification + - forward -## OVModelForCTC +### OVModelForCTC [[autodoc]] openvino.modeling.OVModelForCTC + - forward -## OVModelForAudioXVector +### OVModelForAudioXVector [[autodoc]] openvino.modeling.OVModelForAudioXVector + - forward + +### OVModelForSpeechSeq2Seq + +[[autodoc]] openvino.modeling_seq2seq.OVModelForSpeechSeq2Seq + - forward + + +## Computer Vision -## OVModelForImageClassification +The following classes are available for the following computer vision tasks. + +### OVModelForImageClassification [[autodoc]] openvino.modeling.OVModelForImageClassification + - forward -## OVModelForCausalLM -[[autodoc]] openvino.modeling_decoder.OVModelForCausalLM +## Multimodal -## OVModelForSeq2SeqLM +The following classes are available for the following multimodal tasks. -[[autodoc]] openvino.modeling_seq2seq.OVModelForSeq2SeqLM +### OVModelForVision2Seq + +[[autodoc]] openvino.modeling_seq2seq.OVModelForVision2Seq + - forward + +### OVModelForPix2Struct + +[[autodoc]] openvino.modeling_seq2seq.OVModelForPix2Struct + - forward + +## Custom Tasks + +### OVModelForCustomTasks + +[[autodoc]] openvino.modeling.OVModelForCustomTasks + - forward + +### OVModelForFeatureExtraction + +[[autodoc]] openvino.modeling.OVModelForFeatureExtraction + - forward + + +# Quantization -## OVQuantizer +### OVQuantizer [[autodoc]] openvino.quantization.OVQuantizer