Added stacked tag (#1283)

Author: lu-ohai

ads/aqua/common/enums.py

@@ -44,6 +44,7 @@ class Tags(ExtendedEnum):
  MODEL_FORMAT = "model_format"
  MODEL_ARTIFACT_FILE = "model_file"
  MULTIMODEL_TYPE_TAG = "aqua_multimodel"
+ STACKED_MODEL_TYPE_TAG = "aqua_stacked_model"
  AQUA_FINE_TUNE_MODEL_VERSION = "fine_tune_model_version"
 
 

ads/aqua/model/model.py

@@ -196,6 +196,7 @@ def create(
 
  custom_model = None
  if is_stacked:
+ combined_freeform_tags.update({Tags.STACKED_MODEL_TYPE_TAG: "true"})
  custom_model = self._create_model_group(
  model_id=model,
  model_name=model_name,

ads/aqua/modeldeployment/deployment.py

@@ -947,6 +947,7 @@ def _create(
  params = f"{params} {deployment_params}".strip()
 
  if isinstance(aqua_model, DataScienceModelGroup):
+ tags.update({Tags.STACKED_MODEL_TYPE_TAG: "true"})
  env_var.update({"VLLM_ALLOW_RUNTIME_LORA_UPDATING": "true"})
  env_var.update(
  {"MODEL": f"{AQUA_MODEL_DEPLOYMENT_FOLDER}{aqua_model.base_model_id}/"}