frank-default-small: batch=5120, ubatch=2560
This commit is contained in:
1 parent
635e074af8
commit
e777d3cb56
2 files changed
+5
-5
No files matched your search
@@ -43,9 +43,9 @@ VERBOSITY=4
|
|||||||
# Store SWA/context checkpoints in RAM to hold peak GPU memory down.
|
# Store SWA/context checkpoints in RAM to hold peak GPU memory down.
|
||||||
CTX_CHECKPOINTS=4
|
CTX_CHECKPOINTS=4
|
||||||
|
|
||||||
# Big prefill batches (ubatch 3072 is the top prefill tier).
|
# Big prefill batches (ubatch 2560, batch 5120).
|
||||||
BATCH_SIZE=6144
|
BATCH_SIZE=5120
|
||||||
UBATCH_SIZE=3072
|
UBATCH_SIZE=2560
|
||||||
|
|
||||||
# Eager RAM load (MoE experts on CPU; ~31 GB RAM).
|
# Eager RAM load (MoE experts on CPU; ~31 GB RAM).
|
||||||
NO_MMAP=1
|
NO_MMAP=1
|
||||||
|
|||||||
@@ -38,8 +38,8 @@ fi
|
|||||||
: "${VERBOSITY:=4}"
|
: "${VERBOSITY:=4}"
|
||||||
: "${N_CPU_MOE:=36}"
|
: "${N_CPU_MOE:=36}"
|
||||||
: "${N_GPU_LAYERS:=all}"
|
: "${N_GPU_LAYERS:=all}"
|
||||||
: "${BATCH_SIZE:=6144}"
|
: "${BATCH_SIZE:=5120}"
|
||||||
: "${UBATCH_SIZE:=3072}"
|
: "${UBATCH_SIZE:=2560}"
|
||||||
: "${CTX_CHECKPOINTS:=4}"
|
: "${CTX_CHECKPOINTS:=4}"
|
||||||
: "${NO_MMAP:=1}"
|
: "${NO_MMAP:=1}"
|
||||||
: "${TEMPERATURE:=1.0}"
|
: "${TEMPERATURE:=1.0}"
|
||||||
|
|||||||
Reference in new issue
Block a user