Commit 748fc7b 1 parent fba9606 commit 748fc7b Copy full SHA for 748fc7b
File tree 1 file changed +5
-1
lines changed
ml-algorithms/src/main/java/org/opensearch/ml/engine
1 file changed +5
-1
lines changed Original file line number Diff line number Diff line change 27
27
28
28
import org .opensearch .core .action .ActionListener ;
29
29
import org .opensearch .ml .common .FunctionName ;
30
+ import org .opensearch .ml .common .model .MLDeploySetting ;
30
31
import org .opensearch .ml .common .model .MLModelConfig ;
31
32
import org .opensearch .ml .common .model .MLModelFormat ;
32
33
import org .opensearch .ml .common .model .QuestionAnsweringModelConfig ;
@@ -70,6 +71,7 @@ public void downloadPrebuiltModelConfig(
70
71
boolean deployModel = registerModelInput .isDeployModel ();
71
72
String [] modelNodeIds = registerModelInput .getModelNodeIds ();
72
73
String modelGroupId = registerModelInput .getModelGroupId ();
74
+ MLDeploySetting mlDeploySetting = registerModelInput .getDeploySetting ();
73
75
try {
74
76
AccessController .doPrivileged ((PrivilegedExceptionAction <Void >) () -> {
75
77
@@ -104,7 +106,9 @@ public void downloadPrebuiltModelConfig(
104
106
.modelNodeIds (modelNodeIds )
105
107
.isHidden (isHidden )
106
108
.modelGroupId (modelGroupId )
107
- .functionName (FunctionName .from ((functionName )));
109
+ .functionName (FunctionName .from ((functionName )))
110
+ .deploySetting (mlDeploySetting );
111
+
108
112
config .entrySet ().forEach (entry -> {
109
113
switch (entry .getKey ().toString ()) {
110
114
case MLRegisterModelInput .MODEL_FORMAT_FIELD :
You can’t perform that action at this time.
0 commit comments