falcon.go 277 Bytes
Newer Older
Michael Yang's avatar
Michael Yang committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
package llm

const (
	falconModelType7B   = 32
	falconModelType40B  = 60
	falconModelType180B = 80
)

func falconModelType(numLayer uint32) string {
	switch numLayer {
	case 32:
		return "7B"
	case 60:
		return "40B"
	case 80:
		return "180B"
	default:
Michael Yang's avatar
Michael Yang committed
18
		return "unknown"
Michael Yang's avatar
Michael Yang committed
19
20
	}
}