From 20aaf30785f76af49e6443a49c4b14562c8f2fbe Mon Sep 17 00:00:00 2001
From: CaIon <1808837298@qq.com>
Date: Wed, 24 Apr 2024 18:53:21 +0800
Subject: [PATCH] feat: update model ratio

---
 common/model-ratio.go | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/common/model-ratio.go b/common/model-ratio.go
index ba9c71c..c800fa1 100644
--- a/common/model-ratio.go
+++ b/common/model-ratio.go
@@ -224,6 +224,9 @@ func GetCompletionRatio(name string) float64 {
 	if strings.HasPrefix(name, "gemini-") {
 		return 3
 	}
+	if strings.HasPrefix(name, "command") {
+		return 5
+	}
 	switch name {
 	case "llama2-70b-4096":
 		return 0.8 / 0.7