{
  "adm_in_channels": 2816,
  "model_channels": 320,
  "channel_mults": [1, 2, 4], 
  "num_head_channels": 64,
  "transformer_depths": [1, 2, 10], 
  "context_dim": 2048, 
  "is_refiner": false
}