| { | |
| "_name_or_path": "asRobotics/prosoro-mvae", | |
| "architectures": ["MVAE"], | |
| "model_type": "mvae", | |
| "prosoro_type": "cylinder", | |
| "x_dim_dict": [6, 6, 1749], | |
| "h1_dim_dict": [8, 8, 1024], | |
| "h2_dim_dict": [16, 16, 256], | |
| "z_dim": 32, | |
| "torch_dtype": "float32", | |
| "layer_norm": false, | |
| "use_activation": "relu", | |
| "auto_map": { | |
| "AutoConfig": "modeling.MVAEConfig", | |
| "AutoModel": "modeling.MVAE" | |
| } | |
| } | |