frank-default-small: batch=5120, ubatch=2560

This commit is contained in:
Frank committed 2026-08-20 15:38:18 -07:00
1 parent 635e074af8
commit e777d3cb56
2 files changed
+5 -5

No files matched your search

+3 -3
View File
@@ -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
+2 -2
View File
@@ -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}"