diff --git a/relay/relay-image.go b/relay/relay-image.go index 85cbf2f9..afa5b8e2 100644 --- a/relay/relay-image.go +++ b/relay/relay-image.go @@ -115,8 +115,8 @@ func ImageHelper(c *gin.Context) *dto.OpenAIErrorWithStatusCode { } } - imageRatio := priceData.ModelPrice * sizeRatio * qualityRatio * float64(imageRequest.N) - quota := int(imageRatio * priceData.GroupRatio * common.QuotaPerUnit) + priceData.ModelPrice *= sizeRatio * qualityRatio * float64(imageRequest.N) + quota := int(priceData.ModelPrice * priceData.GroupRatio * common.QuotaPerUnit) if userQuota-quota < 0 { return service.OpenAIErrorWrapperLocal(fmt.Errorf("image pre-consumed quota failed, user quota: %s, need quota: %s", common.FormatQuota(userQuota), common.FormatQuota(quota)), "insufficient_user_quota", http.StatusForbidden)