Add prefixes to image models to solve the problem of duplicate models

2025-11-18 06:03:42 +08:00 · 2024-05-26 00:18:18 +08:00
parent 0acee9a065
commit c81d7204f4
2 changed files with 10 additions and 1 deletions
--- a/relay/billing/ratio/image.go
+++ b/relay/billing/ratio/image.go
@@ -49,3 +49,8 @@ var ImagePromptLengthLimitations = map[string]int{
 	"wanx-v1":                   4000,
 	"cogview-3":                 833,
 }
+
+var ImageOriginModelName = map[string]string{
+	"ali-stable-diffusion-xl":   "stable-diffusion-xl",
+	"ali-stable-diffusion-v1.5": "stable-diffusion-v1.5",
+}