MoS-DFlash-Routers / schema /router_config.schema.json
ryan-0608's picture
hf3-request-level-v1: metadata, configs, code, and registry (#1)
29d1f68
Raw
History Blame Contribute Delete
2.23 kB
{
"$schema": "https://json-schema.org/draft/2020-12/schema",
"$id": "https://huggingface.co/ryan-0608/MoS-DFlash-Routers/raw/main/schema/router_config.schema.json",
"title": "MoS DFlash request-level router config",
"type": "object",
"required": [
"schema_version",
"artifact_id",
"router_kind",
"status",
"target_model",
"domain_outputs",
"feature_contract",
"head",
"weights"
],
"properties": {
"schema_version": {"const": "1.0"},
"artifact_id": {"type": "string", "minLength": 1},
"router_kind": {"const": "request-level"},
"status": {"const": "experimental"},
"target_model": {
"type": "object",
"required": ["repo_id", "revision"],
"properties": {
"repo_id": {"type": "string", "minLength": 1},
"revision": {"type": "string", "pattern": "^[0-9a-f]{40}$"}
},
"additionalProperties": false
},
"domain_outputs": {
"type": "array",
"minItems": 5,
"maxItems": 5,
"items": {
"type": "object",
"required": ["domain_id", "label", "expert_id"],
"properties": {
"domain_id": {"type": "integer", "minimum": 0, "maximum": 4},
"label": {"type": "string"},
"expert_id": {"type": "string"}
},
"additionalProperties": false
}
},
"feature_contract": {"type": "object"},
"head": {
"type": "object",
"required": ["input_dim", "hidden_dim", "output_dim", "dropout"],
"properties": {
"input_dim": {"type": "integer", "minimum": 1},
"hidden_dim": {"type": "integer", "minimum": 1},
"output_dim": {"const": 5},
"dropout": {"type": "number", "minimum": 0, "maximum": 1}
},
"additionalProperties": true
},
"weights": {
"type": "object",
"required": ["filename", "format", "sha256", "size_bytes"],
"properties": {
"filename": {"const": "router_head.safetensors"},
"format": {"const": "safetensors"},
"sha256": {"type": "string", "pattern": "^[0-9a-f]{64}$"},
"size_bytes": {"type": "integer", "minimum": 1}
},
"additionalProperties": true
}
},
"additionalProperties": true
}