Skip to content
This repository was archived by the owner on Jan 10, 2025. It is now read-only.

Commit cce4257

Browse files
committed
add 8-bit version
1 parent 77161f0 commit cce4257

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

gpt2/download.gradle

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -4,9 +4,10 @@ task downloadLiteModel {
44
def downloadFiles = [
55
"https://s3.amazonaws.com/models.huggingface.co/bert/gpt2-vocab.json": "gpt2-vocab.json",
66
"https://s3.amazonaws.com/models.huggingface.co/bert/gpt2-merges.txt": "gpt2-merges.txt",
7-
"https://s3.amazonaws.com/models.huggingface.co/bert/gpt2-fp16-64.tflite": "model.tflite",
8-
// "https://s3.amazonaws.com/models.huggingface.co/bert/gpt2-64.tflite": "model.tflite",
9-
// "https://s3.amazonaws.com/models.huggingface.co/bert/distilgpt2-64.tflite": "model.tflite",
7+
"https://s3.amazonaws.com/models.huggingface.co/bert/gpt2-64-fp16.tflite": "model.tflite", // FP16 quantization version
8+
// "https://s3.amazonaws.com/models.huggingface.co/bert/gpt2-64-8bits.tflite": "model.tflite", // 8-bit integers version
9+
// "https://s3.amazonaws.com/models.huggingface.co/bert/gpt2-64.tflite": "model.tflite", // "raw" version
10+
// "https://s3.amazonaws.com/models.huggingface.co/bert/distilgpt2-64.tflite": "model.tflite", // Distilled version
1011
]
1112
downloadFiles.each { key, value ->
1213
download {

0 commit comments

Comments
 (0)