diff --git a/pyproject.toml b/pyproject.toml index 3dcc1c22..92e54959 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,6 +1,6 @@ [project] name = "celeste-ai" -version = "0.10.0" +version = "0.10.1" description = "Open source, type-safe primitives for multi-modal AI. All capabilities, all providers, one interface" authors = [{name = "Kamilbenkirane", email = "kamil@withceleste.ai"}] readme = "README.md" diff --git a/src/celeste/modalities/images/providers/byteplus/models.py b/src/celeste/modalities/images/providers/byteplus/models.py index fe220cdb..8d5b0fa5 100644 --- a/src/celeste/modalities/images/providers/byteplus/models.py +++ b/src/celeste/modalities/images/providers/byteplus/models.py @@ -17,17 +17,33 @@ ImageParameter.ASPECT_RATIO: Dimensions( min_pixels=1280 * 720, # 921,600 max_pixels=4096 * 4096, # 16,777,216 - min_aspect_ratio=1 / 16, # 0.0625 + min_aspect_ratio=1 / 16, max_aspect_ratio=16, presets={ - "Square 2K": "2048x2048", - "Square 4K": "4096x4096", - "HD 16:9": "1920x1080", - "2K 16:9": "2560x1440", - "4K 16:9": "3840x2160", - "Portrait HD": "1080x1920", - "Portrait 2K": "1440x2560", - "Ultra-wide 21:9": "3024x1296", + "1K 1:1": "1024x1024", + "1K 3:4": "864x1152", + "1K 4:3": "1152x864", + "1K 16:9": "1312x736", + "1K 9:16": "736x1312", + "1K 2:3": "832x1248", + "1K 3:2": "1248x832", + "1K 21:9": "1568x672", + "2K 1:1": "2048x2048", + "2K 3:4": "1728x2304", + "2K 4:3": "2304x1728", + "2K 16:9": "2848x1600", + "2K 9:16": "1600x2848", + "2K 3:2": "2496x1664", + "2K 2:3": "1664x2496", + "2K 21:9": "3136x1344", + "4K 1:1": "4096x4096", + "4K 3:4": "3520x4704", + "4K 4:3": "4704x3520", + "4K 16:9": "5504x3040", + "4K 9:16": "3040x5504", + "4K 2:3": "3328x4992", + "4K 3:2": "4992x3328", + "4K 21:9": "6240x2656", }, ), ImageParameter.QUALITY: Choice(options=["1K", "2K", "4K"]), @@ -44,21 +60,64 @@ ImageParameter.ASPECT_RATIO: Dimensions( min_pixels=2560 * 1440, # 3,686,400 max_pixels=4096 * 4096, # 16,777,216 - min_aspect_ratio=1 / 16, # 0.0625 + min_aspect_ratio=1 / 16, max_aspect_ratio=16, presets={ - "Square 2K": "2048x2048", - "Square 4K": "4096x4096", - "HD 16:9": "1920x1080", - "2K 16:9": "2560x1440", - "4K 16:9": "3840x2160", - "Portrait HD": "1080x1920", - "Portrait 2K": "1440x2560", - "Ultra-wide 21:9": "3024x1296", + "2K 1:1": "2048x2048", + "2K 3:4": "1728x2304", + "2K 4:3": "2304x1728", + "2K 16:9": "2848x1600", + "2K 9:16": "1600x2848", + "2K 3:2": "2496x1664", + "2K 2:3": "1664x2496", + "2K 21:9": "3136x1344", + "4K 1:1": "4096x4096", + "4K 3:4": "3520x4704", + "4K 4:3": "4704x3520", + "4K 16:9": "5504x3040", + "4K 9:16": "3040x5504", + "4K 2:3": "3328x4992", + "4K 3:2": "4992x3328", + "4K 21:9": "6240x2656", }, ), ImageParameter.QUALITY: Choice(options=["2K", "4K"]), ImageParameter.WATERMARK: Bool(), }, ), + Model( + id="seedream-5-0-260128", + provider=Provider.BYTEPLUS, + display_name="Seedream 5.0 Lite", + operations={Modality.IMAGES: {Operation.GENERATE}}, + streaming=True, + parameter_constraints={ + ImageParameter.ASPECT_RATIO: Dimensions( + min_pixels=2560 * 1440, # 3,686,400 + max_pixels=10_404_496, # 3072 * 3072 * 1.1025 + min_aspect_ratio=1 / 16, + max_aspect_ratio=16, + presets={ + "2K 1:1": "2048x2048", + "2K 3:4": "1728x2304", + "2K 4:3": "2304x1728", + "2K 16:9": "2848x1600", + "2K 9:16": "1600x2848", + "2K 3:2": "2496x1664", + "2K 2:3": "1664x2496", + "2K 21:9": "3136x1344", + "3K 1:1": "3072x3072", + "3K 3:4": "2592x3456", + "3K 4:3": "3456x2592", + "3K 16:9": "4096x2304", + "3K 9:16": "2304x4096", + "3K 2:3": "2496x3744", + "3K 3:2": "3744x2496", + "3K 21:9": "4704x2016", + }, + ), + ImageParameter.QUALITY: Choice(options=["2K", "3K"]), + ImageParameter.WATERMARK: Bool(), + }, + ), ]