Bug in LLaVA main.py

[senp98]

Thank you for your excellent work! I noticed a small inconsistency in Line 96 of API/API_LLaVA/main.py, where the argument is defined as:

parser.add_argument("--model", type=str, default="13b")

However, when running the following command:

python main.py \
  --dataset mmvet \  
  --range 0 100 \ 
  --model_name 13b \ 
  --layer_index 22 \  
  --output_folder "../../experiments" \
  --interpolate_method_name LANCZOS \ 
  --enhance_coe 5 \
  --kernel_size 3 \
  --grayscale 0

an error occurs due to the mismatch between --model and --model_name. To avoid this issue, it would be helpful to align the argument name in the script with the one used in the command. I hope you can consider updating it. Thanks again!