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

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