fix: add falcon.go

This commit is contained in:
Michael Yang 2023-09-13 14:47:32 -07:00
parent 949553db23
commit d028853879

22
llm/falcon.go Normal file
View file

@ -0,0 +1,22 @@
package llm
const ModelFamilyFalcon = "falcon"
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:
return "Unknown"
}
}