kfickel/hokusai
1
0
Fork 0
Code Issues 1 Pull requests 1 Projects Releases Packages Wiki Activity Actions

chore: slim down model capabilities
Some checks failed
Continuous Integration / Build Package (push) Successful in 32s
Details
Continuous Integration / Lint, Check & Test (push) Failing after 42s
Details

Browse source
This commit is contained in:
Konstantin Fickel 2026-02-15 10:29:47 +01:00
parent e7270a118d
commit dc6a75f5c4
Signed by: kfickel
GPG key ID: A793722F9933C1A5
1 changed files with 7 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

17
bulkgen/providers/models.py
View file

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