chore: slim down model capabilities

bulkgen/providers/models.py

@@ -13,13 +13,8 @@ class Capability(StrEnum):
     TEXT_GENERATION = "text generation"
     VISION = "vision"
     TEXT_TO_IMAGE = "text-to-image"
-    HIGH_RESOLUTION = "high resolution"
     REFERENCE_IMAGES = "reference images"
     CONTROL_IMAGES = "control images"
-    EDGE_DETECTION = "edge detection"
-    DEPTH_MAP = "depth map"
-    INPAINTING = "inpainting"
-    OUTPAINTING = "outpainting"
 
 
 @dataclass(frozen=True)
@@ -82,7 +77,7 @@ IMAGE_MODELS: list[ModelInfo] = [
         name="flux-pro-1.1-ultra",
         provider="BlackForestLabs",
         type="image",
-        capabilities=[Capability.TEXT_TO_IMAGE, Capability.HIGH_RESOLUTION],
+        capabilities=[Capability.TEXT_TO_IMAGE],
     ),
     ModelInfo(
         name="flux-2-pro",
@@ -103,7 +98,6 @@ IMAGE_MODELS: list[ModelInfo] = [
         capabilities=[
             Capability.TEXT_TO_IMAGE,
             Capability.CONTROL_IMAGES,
-            Capability.EDGE_DETECTION,
         ],
     ),
     ModelInfo(
@@ -113,20 +107,23 @@ IMAGE_MODELS: list[ModelInfo] = [
         capabilities=[
             Capability.TEXT_TO_IMAGE,
             Capability.CONTROL_IMAGES,
-            Capability.DEPTH_MAP,
         ],
     ),
     ModelInfo(
         name="flux-pro-1.0-fill",
         provider="BlackForestLabs",
         type="image",
-        capabilities=[Capability.TEXT_TO_IMAGE, Capability.INPAINTING],
+        capabilities=[
+            Capability.TEXT_TO_IMAGE,
+        ],
     ),
     ModelInfo(
         name="flux-pro-1.0-expand",
         provider="BlackForestLabs",
         type="image",
-        capabilities=[Capability.TEXT_TO_IMAGE, Capability.OUTPAINTING],
+        capabilities=[
+            Capability.TEXT_TO_IMAGE,
+        ],
     ),
 ]