azureml-registry-tools 0.1.0a6__py3-none-any.whl → 0.1.0a8__py3-none-any.whl

azureml/registry/data/model.schema.json

@@ -0,0 +1,597 @@
+{
+  "$schema": "http://json-schema.org/draft-07/schema#",
+  "title": "Azure ML Model Spec Schema",
+  "description": "Schema for validating Azure ML model specification files",
+  "type": "object",
+  "additionalProperties": false,
+  "properties": {
+    "$schema": {
+      "description": "URI reference to the JSON schema",
+      "pattern": "^https://azuremlschemas.azureedge.net/latest/model.schema.json",
+      "type": "string"
+    },
+    "name": {
+      "description": "Machine name of the model (must match what's in the API)",
+      "type": "string"
+    },
+    "path": {
+      "description": "Path to the model files",
+      "type": "string"
+    },
+    "properties": {
+      "description": "Model properties",
+      "type": "object",
+      "additionalProperties": false,
+      "properties": {
+        "languages": {
+          "description": "Not used anymore. Use languages in the tags section",
+          "type": "string"
+        },
+        "datasets": {
+          "description": "Not used anymore. Use training_datasets in the tags section",
+          "type": "string"
+        },
+        "AzureOpenAI-ModelName": {
+          "description": "Name of the Azure OpenAI model name in the cogsvc catalog",
+          "type": "string"
+        },
+        "AzureOpenAI-Version": {
+          "description": "version of the Azure OpenAI model name in the cogsvc catalog",
+          "oneOf": [{ "type": "string" }, { "type": "integer", "minimum": 1 }]
+        },
+        "azureml.copyRegistryFilesToWorkspace": {
+          "description": "Whether registry files are copied to workspace. If false remove the key.",
+          "type": "boolean"
+        },
+        "baseModelWeightsVersion": {
+          "oneOf": [{ "type": "string" }, { "type": "number", "minimum": 1.0 }]
+        },
+        "baseWeightsId": {
+          "type": "string",
+          "enum": [
+            "Meta-Llama-3-8B",
+            "Meta-Llama-3.1-8B",
+            "Meta-Llama-3.1-8B-Instruct"
+          ]
+        },
+        "engineControllerManifestPath": {
+          "description": "Path to the engine controller manifest",
+          "type": "string",
+          "pattern": "^(azureml/|models/|model/)"
+        },
+        "finetune-recommended-sku": {
+          "description": "Skus that can be used for finetune maap",
+          "$ref": "#/definitions/skuListDefinition"
+        },
+        "intellectualPropertyPublisher": {
+          "description": "Intellectual property publisher name",
+          "type": "string"
+        },
+        "SHA": {
+          "description": "SHA hash of the model",
+          "type": "string"
+        },
+        "SharedComputeCapacityEnabled": {
+          "description": "Whether shared compute capacity is enabled",
+          "type": "boolean"
+        },
+        "evaluation-min-sku-spec": {
+          "description": "Minimum SKU specification for evaluation in the format of 'cores|GPUs|RAM(GB)|VRAM(GB)'",
+          "type": "string",
+          "pattern": "^\\d+\\|\\d+\\|\\d+\\|\\d+$"
+        },
+        "evaluation-recommended-sku": {
+          "description": "Evaluation recommended sku",
+          "$ref": "#/definitions/skuListDefinition"
+        },
+        "finetune-min-sku-spec": {
+          "description": "Minimum SKU specification for finetune in the format of 'cores|GPUs|RAM(GB)|VRAM(GB)'",
+          "type": "string",
+          "pattern": "^\\d+\\|\\d+\\|\\d+\\|\\d+$"
+        },
+        "inference-min-sku-spec": {
+          "description": "Minimum SKU specification for inference in the format of 'cores|GPUs|RAM(GB)|VRAM(GB)'",
+          "type": "string",
+          "pattern": "^\\d+\\|\\d+\\|\\d+\\|\\d+$"
+        },
+        "inference-recommended-sku": {
+          "description": "A list of comma separated skus that the client should use to deploy the model",
+          "$ref": "#/definitions/skuListDefinition"
+        },
+        "finetuning-tasks": {
+          "description": "Tasks for which the model can be fine-tuned",
+          "$ref": "#/definitions/finetuningTasksListDefinition"
+        },
+        "model_id": {
+          "description": "Model id",
+          "type": "string",
+          "pattern": "^(https://visionnotebooksdata.blob.core.windows.net)"
+        },
+        "pipeManifestPath": {
+          "description": "Pipe manifest path. Used for OAI models",
+          "type": "string",
+          "pattern": "^(azureml/)"
+        }
+      }
+    },
+    "tags": {
+      "description": "Model tags",
+      "type": "object",
+      "additionalProperties": false,
+      "properties": {
+        "Featured": {
+          "description": "Whether the model is featured",
+          "type": "string",
+          "enum": [""]
+        },
+        "FinetuneDeprecationDate": {
+          "description": "The date that finetune will be deprecated. Date format yyyy-MM-dd. The deprecation will happen at midnight UTC",
+          "type": "string",
+          "pattern": "^\\d{4}-\\d{2}-\\d{2}$"
+        },
+        "FinetuneRetirementDate": {
+          "description": "The date that finetune will be retired. Date format yyyy-MM-dd. The retirement will happen at midnight UTC",
+          "type": "string",
+          "pattern": "^\\d{4}-\\d{2}-\\d{2}$"
+        },
+        "InferenceDeprecationDate": {
+          "description": "The date that inference will be deprecated. Date format yyyy-MM-dd. The deprecation will happen at midnight UTC",
+          "type": "string",
+          "pattern": "^\\d{4}-\\d{2}-\\d{2}$"
+        },
+        "InferenceLegacyDate": {
+          "description": "The date that inference will become legacy. Date format yyyy-MM-dd. The change will happen at midnight UTC",
+          "type": "string",
+          "pattern": "^\\d{4}-\\d{2}-\\d{2}$"
+        },
+        "InferenceRetirementDate": {
+          "description": "The date that inference will be retired. Date format yyyy-MM-dd. The retirement will happen at midnight UTC",
+          "type": "string",
+          "pattern": "^\\d{4}-\\d{2}-\\d{2}$"
+        },
+        "Preview": {
+          "description": "Whether the model is in preview. Remove the key if not in preview.",
+          "type": "string",
+          "enum": [""]
+        },
+        "_aml_system_vanity_registry": {
+          "description": "AML vanity registry",
+          "type": "string",
+          "enum": ["azureml-phi"]
+        },
+        "author": {
+          "description": "Model provider/author name",
+          "type": "string"
+        },
+        "azureOffers": {
+          "description": "Supported azure offer types as a comma-separated string",
+          "type": "string",
+          "pattern": "^(standard-paygo|RestrictedAccess|PTU|batch-paygo|VM)(?:\\s*,\\s*(standard-paygo|RestrictedAccess|PTU|batch-paygo|VM))*$"
+        },
+        "benchmark": {
+          "description": "Benchmark type",
+          "type": "string",
+          "enum": ["quality", "embeddings"]
+        },
+        "datasets": {
+          "description": "Where data was sourced for training as a comma-separated string",
+          "type": "string",
+          "pattern": "^(Publicly available sources|bookcorpus|wikipedia)(?:\\s*,\\s*(Publicly available sources|bookcorpus|wikipedia))*$"
+        },
+        "deploymentOptions": {
+          "description": "Where data was sourced for training as a comma-separated string",
+          "type": "string",
+          "pattern": "^(ServerlessMaaS|UnifiedEndpointMaaS|MaaP|AOAI|Restricted|MonetizedMaaP)(?:\\s*,\\s*(ServerlessMaaS|UnifiedEndpointMaaS|MaaP|AOAI|Restricted|MonetizedMaaP))*$"
+        },
+        "disable-batch": {
+          "description": "Whether batch processing is disabled",
+          "type": "string",
+          "enum": ["true", "false"]
+        },
+        "disable-maap": {
+          "description": "Whether MAAP is disabled",
+          "type": "string",
+          "enum": ["true", "false"]
+        },
+        "displayName": {
+          "description": "Display name for marketing materials",
+          "type": "string"
+        },
+        "evaluation": {
+          "description": "Reference to evaluation documentation",
+          "type": "string"
+        },
+        "freePlayground": {
+          "description": "Whether the model is available in free playground",
+          "type": "string",
+          "enum": ["true", "false"]
+        },
+        "hiddenlayerscanned": {
+          "description": "If the model has been scanned by hiddenlayer. The value is set to \"\" if scanned, otherwise the key should be removed.",
+          "type": "string",
+          "enum": [""]
+        },
+        "huggingface_model_id": {
+          "description": "Id of the model in the hugging face catalog",
+          "type": "string",
+          "pattern": "^(databricks/|google/|codellama/|microsoft/)"
+        },
+        "industry": {
+          "description": "The industry for the model",
+          "type": "string",
+          "enum": [
+            "consumer-goods",
+            "health-and-life-sciences",
+            "manufacturing",
+            "mobility",
+            "financial-services"
+          ]
+        },
+        "inputModalities": {
+          "description": "Input modalities supported (e.g., text, image) as a comma-separated string",
+          "type": "string",
+          "pattern": "^(audio|csv|embeddings|image|json|pdf|text|video)(?:\\s*,\\s*(audio|csv|embeddings|image|json|pdf|text|video))*$"
+        },
+        "inference_supported_envs": {
+          "description": "Supported inference environments",
+          "type": "array",
+          "items": {
+            "type": "string",
+            "enum": ["ds_mii", "hf", "vllm"]
+          }
+        },
+        "invisibleUntil": {
+          "description": "Date until which the model is invisible",
+          "type": "string",
+          "pattern": "^\\d{4}-\\d{2}-\\d{2}T\\d{2}:\\d{2}:\\d{2}\\.\\d{1,6}Z$"
+        },
+        "isDirectFromAzure": {
+          "description": "If the model is an Azure Direct model",
+          "type": "string",
+          "enum": ["true"]
+        },
+        "keywords": {
+          "description": "List of key words as a comma-separated string. Max 3 keywords",
+          "type": "string",
+          "pattern": "^(Agents|Audio|Coding|Conversation|Instruction|Large context|Low latency|Multilingual|Multimodal|Multipurpose|RAG|Reasoning|Summarization|Translation|Understanding|Vision)(?:\\s*,\\s*(Agents|Audio|Coding|Conversation|Instruction|Large context|Low latency|Multilingual|Multimodal|Multipurpose|RAG|Reasoning|Summarization|Translation|Understanding|Vision)){0,2}$"
+        },
+        "languages": {
+          "description": "Comma-separated list of supported languages (ISO 639-1 language codes)",
+          "type": "string",
+          "pattern": "^(aa|ab|ae|af|ak|am|an|ar|as|av|ay|az|ba|be|bg|bh|bi|bm|bn|bo|br|bs|ca|ce|ch|co|cr|cs|cu|cv|cy|da|de|dv|dz|ee|el|en|eo|es|et|eu|fa|ff|fi|fj|fo|fr|fy|ga|gd|gl|gn|gu|gv|ha|he|hi|ho|hr|ht|hu|hy|hz|ia|id|ie|ig|ii|ik|io|is|it|iu|ja|jv|ka|kg|ki|kj|kk|kl|km|kn|ko|kr|ks|ku|kv|kw|ky|la|lb|lg|li|ln|lo|lt|lu|lv|mg|mh|mi|mk|ml|mn|mr|ms|mt|my|na|nb|nd|ne|ng|nl|nn|\"no\"|nr|nv|ny|oc|oj|om|or|os|pa|pi|pl|ps|pt|pt-br|qu|rm|rn|ro|ru|rw|sa|sc|sd|se|sg|si|sk|sl|sm|sn|so|sq|sr|ss|st|su|sv|sw|ta|te|tg|th|ti|tk|tl|tn|to|tr|ts|tt|tw|ty|ug|uk|ur|uz|ve|vi|vo|wa|wo|xh|yi|yo|za|zh|zh-cn|zu)(?:\\s*,\\s*(aa|ab|ae|af|ak|am|an|ar|as|av|ay|az|ba|be|bg|bh|bi|bm|bn|bo|br|bs|ca|ce|ch|co|cr|cs|cu|cv|cy|da|de|dv|dz|ee|el|en|eo|es|et|eu|fa|ff|fi|fj|fo|fr|fy|ga|gd|gl|gn|gu|gv|ha|he|hi|ho|hr|ht|hu|hy|hz|ia|id|ie|ig|ii|ik|io|is|it|iu|ja|jv|ka|kg|ki|kj|kk|kl|km|kn|ko|kr|ks|ku|kv|kw|ky|la|lb|lg|li|ln|lo|lt|lu|lv|mg|mh|mi|mk|ml|mn|mr|ms|mt|my|na|nb|nd|ne|ng|nl|nn|\"no\"|nr|nv|ny|oc|oj|om|or|os|pa|pi|pl|ps|pt|pt-br|qu|rm|rn|ro|ru|rw|sa|sc|sd|se|sg|si|sk|sl|sm|sn|so|sq|sr|ss|st|su|sv|sw|ta|te|tg|th|ti|tk|tl|tn|to|tr|ts|tt|tw|ty|ug|uk|ur|uz|ve|vi|vo|wa|wo|xh|yi|yo|za|zh|zh-cn|zu))*$"
+        },
+        "license": {
+          "description": "License type",
+          "type": "string",
+          "enum": [
+            "custom",
+            "other",
+            "llama2",
+            "apache-2.0",
+            "Llama 3.1 Community Licensed",
+            "MIT"
+          ]
+        },
+        "licenseDescription": {
+          "description": "Description of the license terms",
+          "type": "string"
+        },
+        "maas-finetuning": {
+          "description": "Whether MaaS fine-tuning is enabled. True means enabled, if disabled remove the key",
+          "type": "string",
+          "enum": ["True"]
+        },
+        "maas-finetuning-test": {
+          "description": "Used for testing maas-finetuning in INT. Remove the key if false",
+          "type": "string",
+          "enum": ["True"]
+        },
+        "maas-finetuning-deploy-regions": {
+          "description": "Regions where fine-tuned models can be deployed",
+          "items": {
+            "type": "string",
+            "enum": [
+              "eastus2",
+              "eastus",
+              "northcentralus",
+              "westus3",
+              "westus",
+              "southcentralus"
+            ]
+          },
+          "type": "array"
+        },
+        "maas-finetuning-offer": {
+          "description": "Whether MaaS fine-tuning offer is available. Remove tag if unavailable",
+          "type": "string",
+          "enum": ["True"]
+        },
+        "maas-finetuning-regions": {
+          "description": "Regions where fine-tuning is available as a comma-separated string",
+          "type": "string",
+          "pattern": "^(westus3|eastus2|eastus|southcentralus|northcentralus|westus)(?:\\s*,\\s*(westus3|eastus2|eastus|southcentralus|northcentralus|westus))*$"
+        },
+        "maas-inference": {
+          "description": "Whether MaaS inference is enabled. Remove the key if not supported",
+          "type": "string",
+          "enum": ["true"]
+        },
+        "maxOutputTokens": {
+          "description": "Maximum number of output tokens",
+          "type": "integer",
+          "minimum": 1
+        },
+        "model_specific_defaults": {
+          "description": "Model specific defaults",
+          "type": "object"
+        },
+        "modelCapabilities": {
+          "description": "Model capabilities (e.g., agents, assistants) as a comma-separated string",
+          "type": "string",
+          "pattern": "^(agents|assistants|routing|reasoning|streaming|tool-calling)(?:\\s*,\\s*(agents|assistants|routing|reasoning|streaming|tool-calling))*$"
+        },
+        "modelHash": {
+          "description": "Hash of the model",
+          "type": "string"
+        },
+        "notes": {
+          "description": "Reference to notes documentation",
+          "type": "string"
+        },
+        "outputModalities": {
+          "description": "Output modalities supported (e.g., text) as a comma-separated string",
+          "type": "string",
+          "pattern": "^(audio|csv|embeddings|image|json|pdf|text|video)(?:\\s*,\\s*(audio|csv|embeddings|image|json|pdf|text|video))*$"
+        },
+        "playgroundRateLimitTier": {
+          "description": "Rate limit tier for playground",
+          "type": "string",
+          "enum": ["high", "low", "custom", "embeddings"]
+        },
+        "SharedComputeCapacityEnabled": {
+          "description": "Whether shared compute capacity is enabled. An empty string means enabled. If not enabled remove the key",
+          "type": "string",
+          "enum": [""]
+        },
+        "summary": {
+          "description": "Brief description of the model",
+          "type": "string"
+        },
+        "task": {
+          "description": "Tasks supported by the model as a comma-separated string",
+          "type": "string",
+          "pattern": "^(audio-analysis|audio-classification|audio-generation|automatic-speech-recognition|chat-completion|completions|content-filters|content-safety|conversational-ai|custom-extraction|data-generation|document-analysis|document-ingestion|document-translation|embeddings|face-detection|fill-mask|forecasting|image-analysis|image-classification|image-feature-extraction|image-text-to-text|image-to-image|image-to-text|intelligent-content-processing|intelligent-document-processing|optical-character-recognition|protein-sequence-generation|protein-structure-prediction|responses|responsible-ai|summarization|text-analysis|text-analytics|text-classification|text-generation|text-to-image|text-to-speech|time-series-forecasting|translation|speech-to-text|speech-translation|video-analysis|video-generation|video-text-to-text|visual-question-answering|zero-shot-classification|zero-shot-image-classification)(?:\\s*,\\s*(audio-analysis|audio-classification|audio-generation|automatic-speech-recognition|chat-completion|completions|content-filters|content-safety|conversational-ai|custom-extraction|data-generation|document-analysis|document-ingestion|document-translation|embeddings|face-detection|fill-mask|forecasting|image-analysis|image-classification|image-feature-extraction|image-text-to-text|image-to-image|image-to-text|intelligent-content-processing|intelligent-document-processing|optical-character-recognition|protein-sequence-generation|protein-structure-prediction|responses|responsible-ai|summarization|text-analysis|text-analytics|text-classification|text-generation|text-to-image|text-to-speech|time-series-forecasting|translation|speech-to-text|speech-translation|video-analysis|video-generation|video-text-to-text|visual-question-answering|zero-shot-classification|zero-shot-image-classification))*$"
+        },
+        "textContextWindow": {
+          "description": "Context window size",
+          "type": "integer",
+          "minimum": 1
+        },
+        "training_datasets": {
+          "description": "A description of the training datasets",
+          "type": "string"
+        },
+        "finetune_compute_allow_list": {
+          "description": "List of compute SKUs allowed for finetune with maap as a comma-separated string",
+          "type": "array",
+          "items": { "$ref": "#/definitions/skuEnum" }
+        },
+        "inference_compute_allow_list": {
+          "description": "List of compute SKUs allowed for inference as a comma-separated string",
+          "type": "array",
+          "items": { "$ref": "#/definitions/skuEnum" }
+        },
+        "evaluation_compute_allow_list": {
+          "description": "List of compute SKUs allowed for evaluation",
+          "type": "array",
+          "items": { "$ref": "#/definitions/skuEnum" }
+        },
+        "trainingDataDate": {
+          "description": "Training cutoff date",
+          "type": "string",
+          "pattern": "^(January|February|March|April|May|June|July|August|September|October|November|December)\\s+\\d{4}$"
+        },
+        "newNotes": {
+          "description": "Notes for the model. Can be a full string of notes or it can be the name of an md file in the current directory like 'notes.md'",
+          "type": "string"
+        },
+        "inference_environment_asset_id": {
+          "type": "string",
+          "pattern": "azureml://registries/([^/]+)/environments/([^/]+)/versions/([^/]+)"
+        },
+        "boolTag": {
+          "description": "For testing only!!!",
+          "type": "boolean"
+        },
+        "floatTag": {
+          "description": "For testing only!!!",
+          "type": "number"
+        },
+        "inlineTag": {
+          "description": "For testing only!!!",
+          "type": "string"
+        },
+        "intTag": {
+          "description": "For testing only!!!",
+          "type": "integer"
+        },
+        "featureFlags": {
+          "description": "Comma separated list of feature flags for the models to gate usage",
+            "type": "string",
+            "pattern": "^(OpenAI\\.(?:EAGatingTier|EnterpriseTier|1PGatingTier|OneVetGatingTier|1000CUGatingTier|computer-use-preview|Preview\\.202409)|OpenAITest)(?:\\s*,\\s*(OpenAI\\.(?:EAGatingTier|EnterpriseTier|1PGatingTier|OneVetGatingTier|1000CUGatingTier|computer-use-preview|Preview\\.202409)|OpenAITest))*$"
+        }
+      }
+    },
+    "version": {
+      "description": "Model version",
+      "oneOf": [{ "type": "string" }, { "type": "integer", "minimum": 1 }]
+    }
+  },
+  "required": ["$schema", "name", "path", "tags", "version"],
+  "definitions": {
+    "skuListDefinition": {
+      "type": "string",
+      "description": "A comma-separated list of valid SKUs with no duplicates",
+      "pattern": "^(Standard_(DS3_v2|NC12s_v3|NC16as_T4_v3|NC24ads_A100_v4|NC24rs_v3|NC24s_v3|NC40ads_H100_v5|NC48ads_A100_v4|NC4as_T4_v3|NC64as_T4_v3|NC6s_v3|NC80adis_H100_v5|NC8as_T4_v3|NC96ads_A100_v4|ND40rs_v2|ND96amsr_A100_v4|ND96asr_v4|ND96isr_H100_v5|DS12_v2|DS4_v2|DS5_v2|D16a_v4|D16as_v4|D32a_v4|D32as_v4|D48a_v4|D48as_v4|D64a_v4|D64as_v4|D8a_v4|D8as_v4|D96a_v4|D96as_v4|E16s_v3|E2s_v3|E32s_v3|E48s_v3|E4s_v3|E64s_v3|E8s_v3|F16s_v2|F32s_v2|F48s_v2|F4s_v2|F64s_v2|F72s_v2|F8s_v2|FX12mds|FX24mds|FX36mds|FX48mds|FX4mds|NV12s_v3|NV24s_v3|NV48s_v3))(?:\\s*,\\s*Standard_(DS3_v2|NC12s_v3|NC16as_T4_v3|NC24ads_A100_v4|NC24rs_v3|NC24s_v3|NC40ads_H100_v5|NC48ads_A100_v4|NC4as_T4_v3|NC64as_T4_v3|NC6s_v3|NC80adis_H100_v5|NC8as_T4_v3|NC96ads_A100_v4|ND40rs_v2|ND96amsr_A100_v4|ND96asr_v4|ND96isr_H100_v5|DS12_v2|DS4_v2|DS5_v2|D16a_v4|D16as_v4|D32a_v4|D32as_v4|D48a_v4|D48as_v4|D64a_v4|D64as_v4|D8a_v4|D8as_v4|D96a_v4|D96as_v4|E16s_v3|E2s_v3|E32s_v3|E48s_v3|E4s_v3|E64s_v3|E8s_v3|F16s_v2|F32s_v2|F48s_v2|F4s_v2|F64s_v2|F72s_v2|F8s_v2|FX12mds|FX24mds|FX36mds|FX48mds|FX4mds|NV12s_v3|NV24s_v3|NV48s_v3))*$"
+    },
+    "skuEnum": {
+      "enum": [
+        "Standard_DS3_v2",
+        "Standard_NC12s_v3",
+        "Standard_NC16as_T4_v3",
+        "Standard_NC24ads_A100_v4",
+        "Standard_NC24rs_v3",
+        "Standard_NC24s_v3",
+        "Standard_NC40ads_H100_v5",
+        "Standard_NC48ads_A100_v4",
+        "Standard_NC4as_T4_v3",
+        "Standard_NC64as_T4_v3",
+        "Standard_NC6s_v3",
+        "Standard_NC80adis_H100_v5",
+        "Standard_NC8as_T4_v3",
+        "Standard_NC96ads_A100_v4",
+        "Standard_ND40rs_v2",
+        "Standard_ND96amsr_A100_v4",
+        "Standard_ND96asr_v4",
+        "Standard_ND96isr_H100_v5",
+        "Standard_DS12_v2",
+        "Standard_DS4_v2",
+        "Standard_DS5_v2",
+        "Standard_D16a_v4",
+        "Standard_D16as_v4",
+        "Standard_D32a_v4",
+        "Standard_D32as_v4",
+        "Standard_D48a_v4",
+        "Standard_D48as_v4",
+        "Standard_D64a_v4",
+        "Standard_D64as_v4",
+        "Standard_D8a_v4",
+        "Standard_D8as_v4",
+        "Standard_D96a_v4",
+        "Standard_D96as_v4",
+        "Standard_E16s_v3",
+        "Standard_E2s_v3",
+        "Standard_E32s_v3",
+        "Standard_E48s_v3",
+        "Standard_E4s_v3",
+        "Standard_E64s_v3",
+        "Standard_E8s_v3",
+        "Standard_F16s_v2",
+        "Standard_F32s_v2",
+        "Standard_F48s_v2",
+        "Standard_F4s_v2",
+        "Standard_F64s_v2",
+        "Standard_F72s_v2",
+        "Standard_F8s_v2",
+        "Standard_FX12mds",
+        "Standard_FX24mds",
+        "Standard_FX36mds",
+        "Standard_FX48mds",
+        "Standard_FX4mds",
+        "Standard_NV12s_v3",
+        "Standard_NV24s_v3",
+        "Standard_NV48s_v3"
+      ]
+    },
+    "taskEnum": {
+      "type": "string",
+      "enum": [
+        "audio-analysis",
+        "audio-generation",
+        "automatic-speech-recognition",
+        "chat-completion",
+        "completions",
+        "content-filters",
+        "content-safety",
+        "conversational-ai",
+        "custom-extraction",
+        "data-generation",
+        "document-analysis",
+        "document-ingestion",
+        "document-translation",
+        "embeddings",
+        "face-detection",
+        "fill-mask",
+        "forecasting",
+        "image-analysis",
+        "image-classification",
+        "image-text-to-text",
+        "image-to-image",
+        "image-to-text",
+        "intelligent-content-processing",
+        "intelligent-document-processing",
+        "optical-character-recognition",
+        "protein-sequence-generation",
+        "protein-structure-prediction",
+        "responses",
+        "responsible-ai",
+        "summarization",
+        "text-analysis",
+        "text-analytics",
+        "text-classification",
+        "text-generation",
+        "text-to-image",
+        "text-to-speech",
+        "translation",
+        "speech-to-text",
+        "speech-translation",
+        "video-analysis",
+        "video-generation"
+      ]
+    },
+    "finetuningTasksListDefinition": {
+      "type": "string",
+      "description": "A comma-separated list of valid finetuning tasks with no duplicates",
+      "pattern": "^(automatic-speech-recognition|chat|chat-completion|completions|question-answering|speech-to-text|text-analysis|text-classification|text-generation|text-to-speech)(?:\\s*,\\s*(automatic-speech-recognition|chat|chat-completion|completions|question-answering|speech-to-text|text-analysis|text-classification|text-generation|text-to-speech))*$"
+    },
+    "finetuningTasksEnum": {
+      "type": "string",
+      "enum": [
+        "automatic-speech-recognition",
+        "chat",
+        "chat-completion",
+        "completions",
+        "question-answering",
+        "speech-to-text",
+        "text-analysis",
+        "text-classification",
+        "text-generation",
+        "text-to-speech"
+      ]
+    },
+    "keywordsEnum": {
+      "type": "string",
+      "enum": [
+        "Agents",
+        "Audio",
+        "Coding",
+        "Conversation",
+        "Instruction",
+        "Large context",
+        "Low latency",
+        "Multilingual",
+        "Multimodal",
+        "Multipurpose",
+        "RAG",
+        "Reasoning",
+        "Summarization",
+        "Translation",
+        "Understanding",
+        "Vision"
+      ]
+    },
+    "modalitiesEnum": {
+      "type": "string",
+      "enum": [
+        "audio",
+        "csv",
+        "embeddings",
+        "image",
+        "json",
+        "pdf",
+        "text",
+        "video"
+      ]
+    }
+  }
+}

azureml/registry/data/model.yaml.template

@@ -0,0 +1,8 @@
+path:
+  container_name: your-container-name
+  container_path: your-container-path
+  storage_name: your-storage-name
+  type: azureblob
+publish:
+  description: description.md
+  type: model-type  # Options: custom_model, mlflow_model, triton_model

azureml/registry/data/notes.md.template

@@ -0,0 +1,3 @@
+<!-- DO NOT CHANGE MARKDOWN HEADERS. IF CHANGED, MODEL CARD MAY BE REJECTED BY A REVIEWER -->
+ 
+<!-- `notes.md` is highly recommended, but not required. It captures information about how your model is created. We highly recommend including this section to provide transparency for the customers. -->