Browse Source

fix: add falcon.go

Michael Yang 1 year ago
parent
commit
d028853879
1 changed files with 22 additions and 0 deletions
  1. 22 0
      llm/falcon.go

+ 22 - 0
llm/falcon.go

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