Skip to content

Commit ea5cd4d

Browse files
Merge remote-tracking branch 'origin/extract_share' into support_qwen_phi_gemma_whisper
2 parents 781e953 + 6b30094 commit ea5cd4d

File tree

2 files changed

+10
-23
lines changed

2 files changed

+10
-23
lines changed

examples/mediatek/executor_runner/mtk_llama_executor_runner.cpp

Lines changed: 3 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -70,17 +70,11 @@ DEFINE_string(
7070
token_embedding_path,
7171
"embedding.bin",
7272
"Input token embedding lookup table path.");
73-
DEFINE_string(
74-
prompt_model_paths,
75-
"model_128t.pte",
76-
"Comma-separated prompt model paths.");
77-
DEFINE_string(
78-
gen_model_paths,
79-
"model_1t.pte",
80-
"Comma-separated generative model paths.");
73+
DEFINE_string(prompt_model_paths, "", "Comma-separated prompt model paths.");
74+
DEFINE_string(gen_model_paths, "", "Comma-separated generative model paths.");
8175
DEFINE_string(
8276
model_package_paths,
83-
"model_package.pte",
77+
"",
8478
"Comma-separated weight-shared model package paths.");
8579

8680
// Tokenizer

examples/mediatek/executor_runner/run_llama3_sample.sh

Lines changed: 7 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -36,18 +36,12 @@ TOKENIZER_PATH="/data/local/tmp/llama3/tokenizer.model"
3636
TOKEN_EMBEDDING_PATH="/data/local/tmp/llama3/embedding_llama3_8b_instruct_fp32.bin"
3737

3838
# Comma-Separated Paths
39-
PROMPT_MODEL_PATHS="\
40-
/data/local/tmp/llama3/llama3_8b_SC_sym4W_sym16A_4_chunks_Overall_128t512c_0.pte,\
41-
/data/local/tmp/llama3/llama3_8b_SC_sym4W_sym16A_4_chunks_Overall_128t512c_1.pte,\
42-
/data/local/tmp/llama3/llama3_8b_SC_sym4W_sym16A_4_chunks_Overall_128t512c_2.pte,\
43-
/data/local/tmp/llama3/llama3_8b_SC_sym4W_sym16A_4_chunks_Overall_128t512c_3.pte,"
44-
45-
# Comma-Separated Paths
46-
GEN_MODEL_PATHS="\
47-
/data/local/tmp/llama3/llama3_8b_SC_sym4W_sym16A_4_chunks_Overall_1t512c_0.pte,\
48-
/data/local/tmp/llama3/llama3_8b_SC_sym4W_sym16A_4_chunks_Overall_1t512c_1.pte,\
49-
/data/local/tmp/llama3/llama3_8b_SC_sym4W_sym16A_4_chunks_Overall_1t512c_2.pte,\
50-
/data/local/tmp/llama3/llama3_8b_SC_sym4W_sym16A_4_chunks_Overall_1t512c_3.pte,"
39+
WEIGHT_SHARED_MODEL_PACKAGE_PATHS="\
40+
/data/local/tmp/llama3/llama3_8b_SC_sym4W_sym16A_4_chunks_Overall_0.pte,\
41+
/data/local/tmp/llama3/llama3_8b_SC_sym4W_sym16A_4_chunks_Overall_1.pte,\
42+
/data/local/tmp/llama3/llama3_8b_SC_sym4W_sym16A_4_chunks_Overall_2.pte,\
43+
/data/local/tmp/llama3/llama3_8b_SC_sym4W_sym16A_4_chunks_Overall_3.pte,\
44+
"
5145

5246
PROMPT_FILE=/data/local/tmp/llama3/sample_prompt.txt
5347

@@ -75,6 +69,5 @@ export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$PWD
7569
--tokenizer_type=$TOKENIZER_TYPE \
7670
--tokenizer_path=$TOKENIZER_PATH \
7771
--token_embedding_path=$TOKEN_EMBEDDING_PATH \
78-
--prompt_model_paths=$PROMPT_MODEL_PATHS \
79-
--gen_model_paths=$GEN_MODEL_PATHS \
72+
--model_package_paths=$WEIGHT_SHARED_MODEL_PACKAGE_PATHS \
8073
--prompt_file=$PROMPT_FILE

0 commit comments

Comments
 (0)