File size: 446 Bytes
373af33 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 |
_base_ = ['lmm.py']
model = dict(
model=dict(
latent_part_dim=64,
num_layers=8,
num_cond_layers=2,
dropout=0.1,
ca_block_cfg=dict(
num_experts=16,
topk=4
),
guidance_cfg=dict(
humanml3d_t2m=dict(type='linear', scale=10.5),
),
),
diffusion_test_dict=dict(
humanml3d_t2m='15,15,8,6,6',
),
)
data = dict(samples_per_gpu=32) |