CUDA_VISIBLE_DEVICES="7" accelerate launch \
--num_processes=1 \
--main_process_port 12348 \
-m lmms_eval \
--model internvl2 \
--model_args pretrained="OpenGVLab/InternVL2-8B,device_map=auto" \
--tasks mme \
--batch_size 1 \
--log_samples \
--log_samples_suffix llava_mme \
--output_path ./logs/
|
elif accelerator.num_processes == 1 and device_map == "auto": |
lm model have't
self.accelerator = accelerator, but
lm.accelerator.num_processes attribute will appear in evaluator.py,
|
while len([file for file in os.listdir(cli_args.output_path) if file.endswith('metric_eval_done.txt')]) < lm.accelerator.num_processes: |