config.json

{
  'model': 'NextDiT_2B_GQA_patch2',
  'image_size': 1024,
  'vae': 'sdxl',
  'model_parallel_size': 1,
  'qk_norm': true,
  'lm': 'google/gemma-2b',
  'tokenizer_path': 'google/gemma-2b',
}