alexwengg commited on
Commit
db47ae8
·
verified ·
1 Parent(s): 0d980ef

short-reply-0.6b Core ML package (prefill on ANE, decode GPU), tokenizer, host config

Browse files
.gitattributes CHANGED
@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
 
 
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
36
+ tokenizer.json filter=lfs diff=lfs merge=lfs -text
config.json ADDED
@@ -0,0 +1,14 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "package": "short_reply_0_6b.mlpackage",
3
+ "prefillLength": 160,
4
+ "cacheLength": 512,
5
+ "decodeLengths": [1, 64, 128, 256],
6
+ "layers": 28,
7
+ "kvHeads": 8,
8
+ "headDim": 128,
9
+ "padID": 151643,
10
+ "stopIDs": [151645, 151643],
11
+ "maxNewTokens": 32,
12
+ "systemPrompt": "Write one natural, very short reply to the social post as a separate person. React to a concrete detail in the post when possible. Do not invent facts or personal experiences. Use at most 12 words. Output only the reply.",
13
+ "maxPostCharacters": 2000
14
+ }
short_reply_0_6b.mlpackage/Data/com.apple.CoreML/model.mlmodel ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7d848f2ce5066851927038202d9a929186adee2799fb29feadc652781c9a77b9
3
+ size 1406867
short_reply_0_6b.mlpackage/Data/com.apple.CoreML/weights/weight.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:87eb1dee972ca6533effef3d3431f52f5e59777bf022ef7b39aa55daa907e78f
3
+ size 757896128
short_reply_0_6b.mlpackage/Manifest.json ADDED
@@ -0,0 +1,18 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "fileFormatVersion": "1.0.0",
3
+ "itemInfoEntries": {
4
+ "E1ECD84E-BBE5-44DA-BAD0-26E92E5350D4": {
5
+ "author": "com.apple.CoreML",
6
+ "description": "CoreML Model Weights",
7
+ "name": "weights",
8
+ "path": "com.apple.CoreML/weights"
9
+ },
10
+ "F6B2E484-283D-4DEE-91FD-47F56DE66872": {
11
+ "author": "com.apple.CoreML",
12
+ "description": "CoreML Model Specification",
13
+ "name": "model.mlmodel",
14
+ "path": "com.apple.CoreML/model.mlmodel"
15
+ }
16
+ },
17
+ "rootModelIdentifier": "F6B2E484-283D-4DEE-91FD-47F56DE66872"
18
+ }
tokenizer.json ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:be75606093db2094d7cd20f3c2f385c212750648bd6ea4fb2bf507a6a4c55506
3
+ size 11422650