| { | |
| "model_name": "aam-diffusion-v1.0", | |
| "aam_mind_source": "rsvs_graph", | |
| "aam_body_type": "specialized_diffusion", | |
| "architecture": { | |
| "type": "diffusion_transformer", | |
| "d_model": 64, | |
| "n_layers": 2, | |
| "n_heads": 4, | |
| "d_ff": 128, | |
| "vocab_size": 500, | |
| "max_seq_len": 32, | |
| "pos_encoding_type": "learned" | |
| }, | |
| "diffusion": { | |
| "n_timesteps": 50, | |
| "n_inference_steps": 5, | |
| "schedule_type": "cosine", | |
| "prediction_type": "epsilon", | |
| "sampling_method": "ddim" | |
| }, | |
| "graph_encoder": { | |
| "d_graph": 32, | |
| "n_graph_layers": 1, | |
| "conditioning_method": "cross_attention" | |
| }, | |
| "parameters": 311670 | |
| } |