|
{ |
|
"attn_depth": 12, |
|
"attn_dim_head": 64, |
|
"attn_heads": 12, |
|
"attn_kwargs": { |
|
"attn_num_mem_kv": 4, |
|
"ff_glu": true |
|
}, |
|
"coarse_pre_gateloop_depth": 2, |
|
"condition_on_text": true, |
|
"cross_attn_num_mem_kv": 4, |
|
"dim": 768, |
|
"dropout": 0.0, |
|
"fine_attn_depth": 2, |
|
"fine_attn_dim_head": 32, |
|
"fine_attn_heads": 8, |
|
"fine_cross_attend_text": true, |
|
"fine_pre_gateloop_depth": 2, |
|
"flash_attn": true, |
|
"gateloop_use_heinsen": false, |
|
"max_seq_len": 1500, |
|
"num_sos_tokens": 1, |
|
"pad_id": -1, |
|
"quads": false, |
|
"text_cond_with_film": false, |
|
"text_condition_cond_drop_prob": 0.0, |
|
"text_condition_model_types": "bge" |
|
} |