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