Browse Source

add 34b model type

Michael Yang 1 year ago
parent
commit
b1cececb8e
1 changed files with 3 additions and 0 deletions
  1. 3 0
      llm/ggml.go

+ 3 - 0
llm/ggml.go

@@ -15,6 +15,7 @@ const (
 	ModelType3B  ModelType = 26
 	ModelType7B  ModelType = 32
 	ModelType13B ModelType = 40
+	ModelType34B ModelType = 48
 	ModelType30B ModelType = 60
 	ModelType65B ModelType = 80
 )
@@ -27,6 +28,8 @@ func (mt ModelType) String() string {
 		return "7B"
 	case ModelType13B:
 		return "13B"
+	case ModelType34B:
+		return "34B"
 	case ModelType30B:
 		return "30B"
 	case ModelType65B: