data = dict(
    train=dict(
        type='ImagePrompt',
        data_root='data/t2i_prompts_3m',
        cache_dir='preproc_qwen',
        cache_datalist_path='data/t2i_prompts_3m/preproc_qwen.jsonl.gz',
        prompt_dataset_kwargs=dict(
            path='Lakonik/t2i-prompts-3m',
            split='train'),
        negative_prompt_embeds_path='data/qwen_empty_prompt_embeds.pth',
        pad_seq_len=512,
        bucketize=True,
        end_ind=-128),
    val=dict(
        type='ImagePrompt',
        data_root='data/t2i_prompts_3m',
        cache_dir='preproc_qwen',
        cache_datalist_path='data/t2i_prompts_3m/preproc_qwen.jsonl.gz',
        prompt_dataset_kwargs=dict(
            path='Lakonik/t2i-prompts-3m',
            split='train'),
        negative_prompt_embeds_path='data/qwen_empty_prompt_embeds.pth',
        pad_seq_len=512,
        start_ind=-128,
        repeat=2,
        test_mode=True,
    ),
)