24 lines
336 B
Go
24 lines
336 B
Go
|
package llm
|
||
|
|
|
||
|
|
const (
|
||
|
|
starCoderModelType1B = 24
|
||
|
|
starCoderModelType3B = 36
|
||
|
|
starCoderModelType7B = 42
|
||
|
|
starCoderModelType15B = 40
|
||
|
|
)
|
||
|
|
|
||
|
|
func starCoderModelType(numLayer uint32) string {
|
||
|
|
switch numLayer {
|
||
|
|
case 24:
|
||
|
|
return "1B"
|
||
|
|
case 36:
|
||
|
|
return "3B"
|
||
|
|
case 42:
|
||
|
|
return "7B"
|
||
|
|
case 40:
|
||
|
|
return "15B"
|
||
|
|
default:
|
||
|
|
return "unknown"
|
||
|
|
}
|
||
|
|
}
|