Skip to content

Commit

Permalink
Merge pull request #36 from cocoshe/main
Browse files Browse the repository at this point in the history
[Fix] repr llava doc
  • Loading branch information
Luodian authored May 2, 2024
2 parents b46239c + 8d3526c commit 986139a
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion miscs/repr_scripts.sh
Original file line number Diff line number Diff line change
Expand Up @@ -11,4 +11,4 @@ pip install -r llava_repr_requirements.txt

# Run and exactly reproduce llava_v1.5 results!
# mme as an example
accelerate launch --num_processes=1 -m lmms_eval --model llava --model_args pretrained="liuhaotian/llava-v1.5-7b,use_flash_attention_2=False" --tasks mme --batch_size 1 --log_samples --log_samples_sufix reproduce --output_path ./logs/
accelerate launch --num_processes=1 -m lmms_eval --model llava --model_args pretrained="liuhaotian/llava-v1.5-7b,use_flash_attention_2=False,device_map=auto" --tasks mme --batch_size 1 --log_samples --log_samples_suffix reproduce --output_path ./logs/

0 comments on commit 986139a

Please sign in to comment.