From d028853879dbb5a2eba3f18776931efffaff508d Mon Sep 17 00:00:00 2001
From: Michael Yang <mxyng@pm.me>
Date: Wed, 13 Sep 2023 14:47:32 -0700
Subject: [PATCH] fix: add falcon.go

---
 llm/falcon.go | 22 ++++++++++++++++++++++
 1 file changed, 22 insertions(+)
 create mode 100644 llm/falcon.go

diff --git a/llm/falcon.go b/llm/falcon.go
new file mode 100644
index 00000000..5713b218
--- /dev/null
+++ b/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"
+	}
+}