dh-mc commited on
Commit
18df7a2
·
1 Parent(s): 29e2913

Update eval-shots.sh

Browse files
Files changed (1) hide show
  1. scripts/eval-shots.sh +7 -0
scripts/eval-shots.sh CHANGED
@@ -25,3 +25,10 @@ fi
25
 
26
  echo Evaluating $MODEL_NAME with few-shot learning
27
  python llm_toolkit/eval_shots.py
 
 
 
 
 
 
 
 
25
 
26
  echo Evaluating $MODEL_NAME with few-shot learning
27
  python llm_toolkit/eval_shots.py
28
+
29
+ if [ "$MODEL" == "internlm2_5-7b-chat-1m" ];
30
+ then
31
+ export LOAD_IN_4BIT=true
32
+ export START_NUM_SHOTS=5
33
+ $BASEDIR/scripts/eval-shots.sh internlm internlm2_5-20b-chat
34
+ fi