mirrors2/opencatd-open
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix bug

Browse Source
This commit is contained in:
c菌
2023-11-23 18:30:24 +08:00
parent c838e7be16
commit 68eb6f5ce1
2 changed files with 9 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
pkg/openai/dall-e.go
View File

@@ -51,7 +51,6 @@ func DalleHandler(c *gin.Context) {
model := dalleRequest.Model model := dalleRequest.Model
var chatlog store.Tokens var chatlog store.Tokens
chatlog.Model = model
chatlog.CompletionCount = dalleRequest.N chatlog.CompletionCount = dalleRequest.N
if model == "dall-e" { if model == "dall-e" {
@@ -59,7 +58,7 @@ func DalleHandler(c *gin.Context) {
} }
model = model + "." + dalleRequest.Size model = model + "." + dalleRequest.Size
if model == "dall-e-2" { if dalleRequest.Model == "dall-e-2" || dalleRequest.Model == "dall-e" {
if !slice.Contain([]string{"256x256", "512x512", "1024x1024"}, dalleRequest.Size) { if !slice.Contain([]string{"256x256", "512x512", "1024x1024"}, dalleRequest.Size) {
c.JSON(http.StatusBadRequest, gin.H{ c.JSON(http.StatusBadRequest, gin.H{
"error": gin.H{ "error": gin.H{
@@ -68,7 +67,7 @@ func DalleHandler(c *gin.Context) {
}) })
return return
} }
} else if model == "dall-e-3" { } else if dalleRequest.Model == "dall-e-3" {
if !slice.Contain([]string{"256x256", "512x512", "1024x1024", "1792x1024", "1024x1792"}, dalleRequest.Size) { if !slice.Contain([]string{"256x256", "512x512", "1024x1024", "1792x1024", "1024x1792"}, dalleRequest.Size) {
c.JSON(http.StatusBadRequest, gin.H{ c.JSON(http.StatusBadRequest, gin.H{
"error": gin.H{ "error": gin.H{
@@ -77,8 +76,8 @@ func DalleHandler(c *gin.Context) {
}) })
return return
} }
if dalleRequest.Quality == "HD" { if dalleRequest.Quality == "hd" {
model = model + ".HD" model = model + ".hd"
} }
} else { } else {
c.JSON(http.StatusBadRequest, gin.H{ c.JSON(http.StatusBadRequest, gin.H{
@@ -88,6 +87,7 @@ func DalleHandler(c *gin.Context) {
}) })
return return
} }
chatlog.Model = model
token, _ := c.Get("localuser") token, _ := c.Get("localuser")

8
pkg/tokenizer/tokenizer.go
View File

@@ -116,13 +116,13 @@ func Cost(model string, promptCount, completionCount int) float64 {
cost = float64(0.04 * completion) cost = float64(0.04 * completion)
case "dall-e-3.1024x1792", "dall-e-3.1792x1024": case "dall-e-3.1024x1792", "dall-e-3.1792x1024":
cost = float64(0.08 * completion) cost = float64(0.08 * completion)
case "dall-e-3.256x256.HD": case "dall-e-3.256x256.hd":
cost = float64(0.08 * completion) cost = float64(0.08 * completion)
case "dall-e-3.512x512.HD": case "dall-e-3.512x512.hd":
cost = float64(0.08 * completion) cost = float64(0.08 * completion)
case "dall-e-3.1024x1024.HD": case "dall-e-3.1024x1024.hd":
cost = float64(0.08 * completion) cost = float64(0.08 * completion)
case "dall-e-3.1024x1792.HD", "dall-e-3.1792x1024.HD": case "dall-e-3.1024x1792.hd", "dall-e-3.1792x1024.hd":
cost = float64(0.12 * completion) cost = float64(0.12 * completion)
// claude /million tokens // claude /million tokens