| { |
| "$schema": "https://json-schema.org/draft/2020-12/schema", |
| "title": "Diffusion language baseline experiment registry", |
| "type": "object", |
| "additionalProperties": false, |
| "required": ["step_grids", "models"], |
| "properties": { |
| "step_grids": { |
| "type": "object", |
| "additionalProperties": false, |
| "required": ["many", "few", "fixed_1024"], |
| "properties": { |
| "many": {"const": [1, 2, 4, 8, 16, 32, 1024]}, |
| "few": {"const": [1, 2, 4, 8, 16, 32]}, |
| "fixed_1024": {"const": [1024]} |
| } |
| }, |
| "models": { |
| "type": "object", |
| "additionalProperties": { |
| "type": "object", |
| "additionalProperties": false, |
| "required": ["category", "environment", "adapter", "source", "datasets"], |
| "properties": { |
| "category": {"enum": ["many", "few", "fixed_1024"]}, |
| "step_override": { |
| "type": "array", |
| "minItems": 1, |
| "uniqueItems": true, |
| "items": {"type": "integer", "minimum": 1} |
| }, |
| "environment": {"pattern": "^[a-z][a-z0-9-]*$"}, |
| "adapter": {"pattern": "^[a-z][a-z0-9-]*$"}, |
| "source": {"pattern": "^[a-z][a-z0-9-]*$"}, |
| "datasets": { |
| "type": "object", |
| "additionalProperties": false, |
| "required": ["lm1b", "owt"], |
| "properties": { |
| "lm1b": {"$ref": "#/$defs/dataset_support"}, |
| "owt": {"$ref": "#/$defs/dataset_support"} |
| } |
| } |
| } |
| } |
| } |
| }, |
| "$defs": { |
| "dataset_support": { |
| "type": "object", |
| "additionalProperties": false, |
| "required": ["status"], |
| "properties": { |
| "status": {"enum": ["supported", "unsupported"]}, |
| "provenance": {"enum": ["official", "reference_reproduction", "self_trained"]}, |
| "reason": {"type": "string", "minLength": 1}, |
| "train_recipe": {"type": "string", "pattern": "^[a-z][a-z0-9_]*$"} |
| }, |
| "allOf": [ |
| { |
| "if": {"properties": {"status": {"const": "supported"}}, "required": ["status"]}, |
| "then": { |
| "required": ["provenance"], |
| "not": {"required": ["reason"]} |
| } |
| }, |
| { |
| "if": {"properties": {"status": {"const": "unsupported"}}, "required": ["status"]}, |
| "then": { |
| "required": ["reason"], |
| "not": { |
| "anyOf": [ |
| {"required": ["provenance"]}, |
| {"required": ["train_recipe"]} |
| ] |
| } |
| } |
| } |
| ] |
| } |
| } |
| } |
|
|
