feat: 新增to product img flux2 版,停用sdxl版

fix:
2026-01-26 15:19:28 +08:00
parent 863d9287dc
commit cdaeb6daac
2 changed files with 71 additions and 30 deletions
--- a/app/schemas/generate_image.py
+++ b/app/schemas/generate_image.py
@@ -32,6 +32,13 @@ class GenerateProductImageModel(BaseModel):
    product_type: str


+class Flux2ToProductImgModel(BaseModel):
+    tasks_id: str
+    prompt: str
+    image_url: str
+    image_strength: int
+
+
 class GenerateRelightImageModel(BaseModel):
    tasks_id: str
    prompt: str