philschmid HF staff commited on
Commit
3bdb891
1 Parent(s): 1960ccb

Create gpt2.json

Browse files
Files changed (1) hide show
  1. inference-cache-config/gpt2.json +16 -0
inference-cache-config/gpt2.json ADDED
@@ -0,0 +1,16 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "openai-community/gpt2": [
3
+ {
4
+ "batch_size": 1,
5
+ "sequence_length": 1024,
6
+ "num_cores": 1,
7
+ "auto_cast_type": "fp16"
8
+ },
9
+ {
10
+ "batch_size": 16,
11
+ "sequence_length": 1024,
12
+ "num_cores": 1,
13
+ "auto_cast_type": "fp16"
14
+ }
15
+ ]
16
+ }