From 8d3526c0869f0ad7747ff6bb02441140792b461c Mon Sep 17 00:00:00 2001
From: cocoshe <1228759711@qq.com>
Date: Thu, 28 Mar 2024 13:38:36 +0800
Subject: [PATCH] fix doc

---
 miscs/repr_scripts.sh | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/miscs/repr_scripts.sh b/miscs/repr_scripts.sh
index 27fccbafc..cc9a8ae7d 100644
--- a/miscs/repr_scripts.sh
+++ b/miscs/repr_scripts.sh
@@ -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/
\ No newline at end of file
+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/
\ No newline at end of file