mlc-ai
/

Llama-2-7b-chat-hf-q4f32_1-MLC

Model card Files Files and versions Community

ruihanglai commited on Apr 15

Commit

f4f7e3d

•

1 Parent(s): 3c039fb

Update mlc-chat-config.json

Files changed (1) hide show

mlc-chat-config.json +2 -1

mlc-chat-config.json CHANGED Viewed

@@ -13,7 +13,8 @@
     "prefill_chunk_size": 4096,
     "num_key_value_heads": 32,
     "head_dim": 128,
-    "tensor_parallel_shards": 1
   },
   "vocab_size": 32000,
   "context_window_size": 4096,

     "prefill_chunk_size": 4096,
     "num_key_value_heads": 32,
     "head_dim": 128,
+    "tensor_parallel_shards": 1,
+    "max_batch_size": 80
   },
   "vocab_size": 32000,
   "context_window_size": 4096,