[bhavanatumma/ta2614748] - signature fix

Azure · Mar 7, 2024 · 1f4298a · 1f4298a
1 parent 2dab0dc
commit 1f4298a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/...ts/training/model_management/src/azureml/model/mgmt/processors/transformers/convertors.py b/...ts/training/model_management/src/azureml/model/mgmt/processors/transformers/convertors.py
@@ -500,7 +500,7 @@ def get_model_signature(self):
             outputs = Schema([ColSpec(DataType.string)])
             params = ParamSchema([ParamSpec("top_p", "float", default=1.0),
                                   ParamSpec("temperature", "float", default=0.0),
-                                  ParamSpec("max_new_tokens", "integer", default=250),
+                                  ParamSpec("max_new_tokens", "integer", default=100),
                                   ParamSpec("do_sample", "boolean", default=False),
                                   ParamSpec("return_full_text", "boolean", default=True)])
             return ModelSignature(inputs=inputs, outputs=outputs, params=params)