forked from third-party-mirrors/ollama
65 lines
1.5 KiB
Go
65 lines
1.5 KiB
Go
package format
|
|
|
|
import (
|
|
"fmt"
|
|
"math"
|
|
)
|
|
|
|
const (
|
|
Thousand = 1000
|
|
Million = Thousand * 1000
|
|
Billion = Million * 1000
|
|
Trillion = Billion * 1000
|
|
)
|
|
|
|
func RoundedParameter(b uint64) string {
|
|
switch {
|
|
case b >= Billion:
|
|
number := float64(b) / Billion
|
|
if number == math.Floor(number) {
|
|
return fmt.Sprintf("%.0fB", number) // no decimals if whole number
|
|
}
|
|
return fmt.Sprintf("%.1fB", number) // one decimal if not a whole number
|
|
case b >= Million:
|
|
number := float64(b) / Million
|
|
if number == math.Floor(number) {
|
|
return fmt.Sprintf("%.0fM", number) // no decimals if whole number
|
|
}
|
|
return fmt.Sprintf("%.2fM", number) // two decimals if not a whole number
|
|
case b >= Thousand:
|
|
return fmt.Sprintf("%.0fK", float64(b)/Thousand)
|
|
default:
|
|
return fmt.Sprintf("%d", b)
|
|
}
|
|
}
|
|
|
|
func Parameters(b uint64) string {
|
|
switch {
|
|
case b >= Trillion:
|
|
number := float64(b) / Trillion
|
|
return fmt.Sprintf("%sT", decimalPlace(number))
|
|
case b >= Billion:
|
|
number := float64(b) / Billion
|
|
return fmt.Sprintf("%sB", decimalPlace(number))
|
|
case b >= Million:
|
|
number := float64(b) / Million
|
|
return fmt.Sprintf("%sM", decimalPlace(number))
|
|
case b >= Thousand:
|
|
number := float64(b) / Thousand
|
|
return fmt.Sprintf("%sK", decimalPlace(number))
|
|
default:
|
|
return fmt.Sprintf("%d", b)
|
|
}
|
|
}
|
|
|
|
func decimalPlace(number float64) string {
|
|
switch {
|
|
case number >= 100:
|
|
return fmt.Sprintf("%.0f", number)
|
|
case number >= 10:
|
|
return fmt.Sprintf("%.1f", number)
|
|
default:
|
|
return fmt.Sprintf("%.2f", number)
|
|
}
|
|
}
|