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