@@ -36,18 +36,12 @@ TOKENIZER_PATH="/data/local/tmp/llama3/tokenizer.model"
36
36
TOKEN_EMBEDDING_PATH=" /data/local/tmp/llama3/embedding_llama3_8b_instruct_fp32.bin"
37
37
38
38
# 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
+ "
51
45
52
46
PROMPT_FILE=/data/local/tmp/llama3/sample_prompt.txt
53
47
@@ -75,6 +69,5 @@ export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$PWD
75
69
--tokenizer_type=$TOKENIZER_TYPE \
76
70
--tokenizer_path=$TOKENIZER_PATH \
77
71
--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 \
80
73
--prompt_file=$PROMPT_FILE
0 commit comments