UniMate / config.json
Friedrich-M
Claude Opus 5.5
Release unimate_uniml3d_f60_v3_cross_attn trained to 150k steps with checkpoints 100k-150k and add it to the model card and index; stop distributing training logs and samples for every model
174fc75
Raw History Blame Contribute Delete
3.51 kB
{
"name": "UniMate",
"code": "https://github.com/Friedrich-M/UniMate",
"paper": "https://arxiv.org/abs/2609.05415",
"models": {
"unimate_uniml3d_f60_v3": {
"dataset": "Linzhan/UniML3D",
"train_config": "configs/uniml3d_60frames_graph_adaln.json",
"attention": "graph",
"text_cond": "adaln",
"num_layers": 10,
"latent_dim": 512,
"num_frames": 60,
"fps": 30,
"steps": 150000,
"recommended_checkpoint": "checkpoints/checkpoint_step_150000.pt",
"status": "recommended",
"dataset_revision": "d0f19d943038fb782a26c2a615bd9b3d4f3a8728"
},
"unimate_uniml3d_f60_v3_cross_attn": {
"dataset": "Linzhan/UniML3D",
"train_config": "configs/uniml3d_60frames_full_cross_attn.json",
"attention": "full",
"text_cond": "cross_attn",
"num_layers": 10,
"latent_dim": 512,
"num_frames": 60,
"fps": 30,
"steps": 150000,
"recommended_checkpoint": "checkpoints/checkpoint_step_150000.pt",
"status": "variant",
"dataset_revision": "d0f19d943038fb782a26c2a615bd9b3d4f3a8728"
},
"unimate_uniml3d_f60_v2": {
"dataset": "Linzhan/UniML3D",
"train_config": "unimate_uniml3d_f60_v2/config.json",
"attention": "graph",
"text_cond": "adaln",
"num_layers": 10,
"latent_dim": 512,
"num_frames": 60,
"fps": 30,
"steps": 100000,
"recommended_checkpoint": "checkpoints/checkpoint_step_100000.pt",
"status": "previous",
"dataset_revision": "faaa81773b315247b03f183548e3898dbec2ce80"
},
"unimate_uniml3d_f60_v2_cross_attn": {
"dataset": "Linzhan/UniML3D",
"train_config": "unimate_uniml3d_f60_v2_cross_attn/config.json",
"attention": "full",
"text_cond": "cross_attn",
"num_layers": 10,
"latent_dim": 512,
"num_frames": 60,
"fps": 30,
"steps": 100000,
"recommended_checkpoint": "checkpoints/checkpoint_step_100000.pt",
"status": "previous",
"dataset_revision": "faaa81773b315247b03f183548e3898dbec2ce80"
},
"unimate_mixamo_f60_v2": {
"dataset": "Linzhan/UniML3D",
"train_config": "configs/mixamo_60frames_graph_adaln.json",
"attention": "graph",
"text_cond": "adaln",
"num_layers": 6,
"latent_dim": 512,
"num_frames": 60,
"fps": 30,
"steps": 120000,
"recommended_checkpoint": "checkpoints/checkpoint_step_120000.pt",
"status": "recommended_for_mixamo",
"dataset_revision": "faaa81773b315247b03f183548e3898dbec2ce80"
},
"unimate_truebones_f60_v2": {
"dataset": "Linzhan/UniML3D",
"train_config": "configs/truebones_60frames_graph_adaln.json",
"attention": "graph",
"text_cond": "adaln",
"num_layers": 6,
"latent_dim": 512,
"num_frames": 60,
"fps": 30,
"steps": 80000,
"recommended_checkpoint": "checkpoints/checkpoint_step_80000.pt",
"status": "recommended_for_truebones",
"dataset_revision": "faaa81773b315247b03f183548e3898dbec2ce80"
},
"unimate_uniml3d_f60_v1_preview": {
"dataset": "Linzhan/UniML3D",
"train_config": "unimate_uniml3d_f60_v1_preview/config.json",
"attention": "graph",
"text_cond": "adaln",
"num_layers": 10,
"latent_dim": 512,
"num_frames": 60,
"fps": 30,
"steps": 120000,
"recommended_checkpoint": "checkpoints/checkpoint_step_120000.pt",
"status": "superseded"
}
}
}