Commit
·
2bfe32d
1
Parent(s):
edfbcc2
added new image models
Browse files- frontend/lib/constants/models.ts +5 -0
- services/image.py +25 -0
frontend/lib/constants/models.ts
CHANGED
|
@@ -12,8 +12,13 @@ export const IMAGE_MODELS = [
|
|
| 12 |
{ value: "imagen-4-ultra", label: "Imagen-4 Ultra (Google)" },
|
| 13 |
{ value: "recraft-v3", label: "Recraft V3" },
|
| 14 |
{ value: "ideogram-v3", label: "Ideogram V3" },
|
|
|
|
| 15 |
{ value: "photon", label: "Photon (Luma)" },
|
| 16 |
{ value: "seedream-3", label: "Seedream-3 (ByteDance)" },
|
|
|
|
|
|
|
|
|
|
|
|
|
| 17 |
] as const;
|
| 18 |
|
| 19 |
export type ImageModel = typeof IMAGE_MODELS[number]["value"];
|
|
|
|
| 12 |
{ value: "imagen-4-ultra", label: "Imagen-4 Ultra (Google)" },
|
| 13 |
{ value: "recraft-v3", label: "Recraft V3" },
|
| 14 |
{ value: "ideogram-v3", label: "Ideogram V3" },
|
| 15 |
+
{ value: "ideogram-v3-turbo", label: "Ideogram V3 Turbo" },
|
| 16 |
{ value: "photon", label: "Photon (Luma)" },
|
| 17 |
{ value: "seedream-3", label: "Seedream-3 (ByteDance)" },
|
| 18 |
+
{ value: "seedream-4.5", label: "Seedream-4.5 (ByteDance)" },
|
| 19 |
+
{ value: "flux-2-max", label: "Flux 2 Max (Black Forest Labs)" },
|
| 20 |
+
{ value: "qwen-image", label: "Qwen Image (Pruna AI)" },
|
| 21 |
+
{ value: "p-image", label: "P-Image (Pruna AI)" },
|
| 22 |
] as const;
|
| 23 |
|
| 24 |
export type ImageModel = typeof IMAGE_MODELS[number]["value"];
|
services/image.py
CHANGED
|
@@ -66,6 +66,31 @@ MODEL_REGISTRY: Dict[str, Dict[str, Any]] = {
|
|
| 66 |
"param_name": "aspect_ratio",
|
| 67 |
"uses_dimensions": False,
|
| 68 |
},
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 69 |
}
|
| 70 |
|
| 71 |
# Default model fallback chain (same as original project)
|
|
|
|
| 66 |
"param_name": "aspect_ratio",
|
| 67 |
"uses_dimensions": False,
|
| 68 |
},
|
| 69 |
+
"seedream-4.5": {
|
| 70 |
+
"id": "bytedance/seedream-4.5",
|
| 71 |
+
"param_name": "aspect_ratio",
|
| 72 |
+
"uses_dimensions": False,
|
| 73 |
+
},
|
| 74 |
+
"flux-2-max": {
|
| 75 |
+
"id": "black-forest-labs/flux-2-max",
|
| 76 |
+
"param_name": "aspect_ratio",
|
| 77 |
+
"uses_dimensions": False,
|
| 78 |
+
},
|
| 79 |
+
"qwen-image": {
|
| 80 |
+
"id": "prunaai/z-image-turboqwen/qwen-image",
|
| 81 |
+
"param_name": "aspect_ratio",
|
| 82 |
+
"uses_dimensions": False,
|
| 83 |
+
},
|
| 84 |
+
"p-image": {
|
| 85 |
+
"id": "prunaai/p-image",
|
| 86 |
+
"param_name": "aspect_ratio",
|
| 87 |
+
"uses_dimensions": False,
|
| 88 |
+
},
|
| 89 |
+
"ideogram-v3-turbo": {
|
| 90 |
+
"id": "ideogram-ai/ideogram-v3-turbo",
|
| 91 |
+
"param_name": "aspect_ratio",
|
| 92 |
+
"uses_dimensions": False,
|
| 93 |
+
},
|
| 94 |
}
|
| 95 |
|
| 96 |
# Default model fallback chain (same as original project)
|