Browse Source

add macOS memory check for 47B models

Jeffrey Morgan 1 year ago
parent
commit
77a66df72c
1 changed files with 2 additions and 0 deletions
  1. 2 0
      llm/llm.go

+ 2 - 0
llm/llm.go

@@ -51,6 +51,8 @@ func New(workDir, model string, adapters, projectors []string, opts api.Options)
 			requiredMemory = 16 * format.GigaByte
 		case "30B", "34B", "40B":
 			requiredMemory = 32 * format.GigaByte
+		case "47B":
+			requiredMemory = 48 * format.GigaByte
 		case "65B", "70B":
 			requiredMemory = 64 * format.GigaByte
 		case "180B":