Merge pull request #525 from jmorganca/mxyng/falcon-decode

fix: add falcon.go
This commit is contained in:
Michael Yang 2023-09-13 15:08:47 -07:00 committed by GitHub
commit f89c23764b
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

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"
}
}