@@ -395,10 +395,10 @@ the model definition is not supplied.
395
395
(Required, object)
396
396
The default configuration for inference. This can be: `regression`,
397
397
`classification`, `fill_mask`, `ner`, `question_answering`,
398
- `text_classification`, `text_embedding` or `zero_shot_classification`.
398
+ `text_classification`, `text_embedding`, `text_expansion` or `zero_shot_classification`.
399
399
If `regression` or `classification`, it must match the `target_type` of the
400
400
underlying `definition.trained_model`. If `fill_mask`, `ner`,
401
- `question_answering`, `text_classification`, or `text_embedding `; the
401
+ `question_answering`, `text_classification`, `text_embedding` or `text_expansion `; the
402
402
`model_type` must be `pytorch`.
403
403
+
404
404
.Properties of `inference_config`
@@ -592,6 +592,25 @@ Refer to <<tokenization-properties>> to review the properties of the
592
592
`tokenization` object.
593
593
=====
594
594
595
+ `text_expansion`:::
596
+ (Object, optional)
597
+ include::{es-ref-dir}/ml/ml-shared.asciidoc[tag=inference-config-text-expansion]
598
+ +
599
+ .Properties of text_expansion inference
600
+ [%collapsible%open]
601
+ =====
602
+ `results_field`::::
603
+ (Optional, string)
604
+ include::{es-ref-dir}/ml/ml-shared.asciidoc[tag=inference-config-results-field]
605
+
606
+ `tokenization`::::
607
+ (Optional, object)
608
+ include::{es-ref-dir}/ml/ml-shared.asciidoc[tag=inference-config-nlp-tokenization]
609
+ +
610
+ Refer to <<tokenization-properties>> to review the properties of the
611
+ `tokenization` object.
612
+ =====
613
+
595
614
`text_similarity`:::
596
615
(Object, optional)
597
616
include::{es-ref-dir}/ml/ml-shared.asciidoc[tag=inference-config-text-similarity]
0 commit comments