File tree Expand file tree Collapse file tree 3 files changed +6
-0
lines changed
examples/demo-apps/android/LlamaDemo/app/src/main/java/com/example/executorchllamademo Expand file tree Collapse file tree 3 files changed +6
-0
lines changed Original file line number Diff line number Diff line change 11
11
public enum ModelType {
12
12
LLAMA_3 ,
13
13
LLAMA_3_1 ,
14
+ LLAMA_3_2 ,
14
15
LLAVA_1_5 ,
15
16
LLAMA_GUARD_3 ,
16
17
}
Original file line number Diff line number Diff line change @@ -21,6 +21,7 @@ public static int getModelCategory(ModelType modelType) {
21
21
return VISION_MODEL ;
22
22
case LLAMA_3 :
23
23
case LLAMA_3_1 :
24
+ case LLAMA_3_2 :
24
25
default :
25
26
return TEXT_MODEL ;
26
27
}
Original file line number Diff line number Diff line change @@ -19,6 +19,7 @@ public static String getSystemPromptTemplate(ModelType modelType) {
19
19
switch (modelType ) {
20
20
case LLAMA_3 :
21
21
case LLAMA_3_1 :
22
+ case LLAMA_3_2 :
22
23
return "<|begin_of_text|><|start_header_id|>system<|end_header_id|>\n "
23
24
+ SYSTEM_PLACEHOLDER
24
25
+ "<|eot_id|>" ;
@@ -33,6 +34,7 @@ public static String getUserPromptTemplate(ModelType modelType) {
33
34
switch (modelType ) {
34
35
case LLAMA_3 :
35
36
case LLAMA_3_1 :
37
+ case LLAMA_3_2 :
36
38
case LLAMA_GUARD_3 :
37
39
return "<|start_header_id|>user<|end_header_id|>\n "
38
40
+ USER_PLACEHOLDER
@@ -49,6 +51,7 @@ public static String getConversationFormat(ModelType modelType) {
49
51
switch (modelType ) {
50
52
case LLAMA_3 :
51
53
case LLAMA_3_1 :
54
+ case LLAMA_3_2 :
52
55
return getUserPromptTemplate (modelType ) + "\n " + ASSISTANT_PLACEHOLDER + "<|eot_id|>" ;
53
56
case LLAVA_1_5 :
54
57
return USER_PLACEHOLDER + " ASSISTANT:" ;
@@ -61,6 +64,7 @@ public static String getStopToken(ModelType modelType) {
61
64
switch (modelType ) {
62
65
case LLAMA_3 :
63
66
case LLAMA_3_1 :
67
+ case LLAMA_3_2 :
64
68
case LLAMA_GUARD_3 :
65
69
return "<|eot_id|>" ;
66
70
case LLAVA_1_5 :
You can’t perform that action at this time.
0 commit comments