sushilideaclan01 commited on
Commit
2bfe32d
·
1 Parent(s): edfbcc2

added new image models

Browse files
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)