srun -p s2_bigdata --gres=gpu:0 --kill-on-bad-exit=1 python download.py --repo_id llava_coco_gen_758k