reference/code/flm/configs/model/tiny-dimamba.yaml

name: tiny
type: dimamba
hidden_size: 512
cond_dim: 128
length: 1024
n_blocks: 14
n_heads: 8
scale_by_sigma: True
dropout: 0.1
temb_strategy: adaln 
tie_word_embeddings: False