add macOS memory check for 47B models

This commit is contained in:
Jeffrey Morgan 2024-01-03 19:46:16 -05:00
parent 5b4837f881
commit 77a66df72c

View file

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