Increase medium prefill micro-batch
This commit is contained in:
+1
-1
@@ -26,7 +26,7 @@ FAST_BATCH_SIZE=64
|
|||||||
FAST_UBATCH_SIZE=32
|
FAST_UBATCH_SIZE=32
|
||||||
MEDIUM_CONTEXT=160000
|
MEDIUM_CONTEXT=160000
|
||||||
MEDIUM_BATCH_SIZE=2048
|
MEDIUM_BATCH_SIZE=2048
|
||||||
MEDIUM_UBATCH_SIZE=128
|
MEDIUM_UBATCH_SIZE=512
|
||||||
LARGE_CONTEXT=192000
|
LARGE_CONTEXT=192000
|
||||||
LARGE_BATCH_SIZE=2048
|
LARGE_BATCH_SIZE=2048
|
||||||
LARGE_UBATCH_SIZE=128
|
LARGE_UBATCH_SIZE=128
|
||||||
|
|||||||
+1
-1
@@ -183,7 +183,7 @@ services:
|
|||||||
- --batch-size
|
- --batch-size
|
||||||
- "${MEDIUM_BATCH_SIZE:-2048}"
|
- "${MEDIUM_BATCH_SIZE:-2048}"
|
||||||
- --ubatch-size
|
- --ubatch-size
|
||||||
- "${MEDIUM_UBATCH_SIZE:-128}"
|
- "${MEDIUM_UBATCH_SIZE:-512}"
|
||||||
- --parallel
|
- --parallel
|
||||||
- "${MEDIUM_PARALLEL_SLOTS:-1}"
|
- "${MEDIUM_PARALLEL_SLOTS:-1}"
|
||||||
- --kv-unified
|
- --kv-unified
|
||||||
|
|||||||
@@ -84,7 +84,7 @@ FAST_BATCH_SIZE=64
|
|||||||
FAST_UBATCH_SIZE=32
|
FAST_UBATCH_SIZE=32
|
||||||
MEDIUM_CONTEXT=160000
|
MEDIUM_CONTEXT=160000
|
||||||
MEDIUM_BATCH_SIZE=2048
|
MEDIUM_BATCH_SIZE=2048
|
||||||
MEDIUM_UBATCH_SIZE=128
|
MEDIUM_UBATCH_SIZE=512
|
||||||
MEDIUM_TENSOR_SPLIT=85,15
|
MEDIUM_TENSOR_SPLIT=85,15
|
||||||
LARGE_CONTEXT=192000
|
LARGE_CONTEXT=192000
|
||||||
LARGE_BATCH_SIZE=2048
|
LARGE_BATCH_SIZE=2048
|
||||||
|
|||||||
Reference in New Issue
Block a user