poetry run format

This commit is contained in:
Toran Bruce Richards
2024-09-29 21:02:19 +01:00
parent bf851f371b
commit 75265b16e4

View File

@@ -21,13 +21,15 @@ class ReplicateFluxModelName(str, Enum):
ReplicateFluxModelName.FLUX_DEV: "black-forest-labs/flux-dev",
}
return api_names[self]
# Image type Enum
class ImageType(str, Enum):
WEBP = "webp"
JPG = "jpg"
PNG = "png"
class ReplicateFluxAdvancedModelBlock(Block):
class Input(BlockSchema):
api_key: BlockSecret = SecretField(
@@ -48,7 +50,6 @@ class ReplicateFluxAdvancedModelBlock(Block):
description="Random seed. Set for reproducible generation",
default=None,
title="Seed",
default=None,
)
steps: int = SchemaField(
description="Number of diffusion steps",