llm: Fix debug logging for memory estimates

This commit is contained in:
Jesse Gross 2025-03-25 11:41:26 -07:00 committed by Jesse Gross
parent 1feff61977
commit f4f0992b6e

View File

@ -374,7 +374,7 @@ func (m MemoryEstimate) LogValue() slog.Value {
slog.Group(
"weights",
// memory of the weights
"total", format.HumanBytes2(m.memoryWeights),
"total", format.HumanBytes2(m.memoryWeights+m.memoryLayerOutput),
// memory of repeating layers
"repeating", format.HumanBytes2(m.memoryWeights),
// memory of non-repeating layers