Spaces:
Build error
Build error
Update eval-shots.sh
Browse files- 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
|