diff --git a/run_file/DynamicKV-LLM_Small_Minimind.sh b/run_file/DynamicKV-LLM_Small_Minimind.sh index f55597b..dacedde 100644 --- a/run_file/DynamicKV-LLM_Small_Minimind.sh +++ b/run_file/DynamicKV-LLM_Small_Minimind.sh @@ -46,5 +46,5 @@ CUDA_VISIBLE_DEVICES=0,1,2,3 accelerate launch \ --use_flash_attn \ --profile \ --profile_interval 10\ - --knowlwdge_num 1024 \ - --knowlwdge_length 8 + --knowledge_num 16384 \ + --knowledge_length 64