Skip to content

Commit

Permalink
fix: update docstring for STABLE_DIFFUSION_BASELINE_CATEGORY
Browse files Browse the repository at this point in the history
The tests automatically generation the `.schema` file using this docstring, so it must also be updated.
  • Loading branch information
tazlin committed Sep 11, 2024
1 parent 80465b8 commit 4e01568
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions horde_model_reference/meta_consts.py
Original file line number Diff line number Diff line change
Expand Up @@ -82,15 +82,15 @@ class MODEL_PURPOSE(StrEnum):
miscellaneous = auto()


class STABLE_DIFFUSION_BASELINE_CATEGORY(StrEnum):
"""An enum of all the stable diffusion baselines."""
class STABLE_DIFFUSION_BASELINE_CATEGORY(StrEnum):
"""An enum of all the image generation baselines."""

stable_diffusion_1 = auto()
stable_diffusion_2_768 = auto()
stable_diffusion_2_512 = auto()
stable_diffusion_xl = auto()
stable_cascade = auto()
flux_1 = auto()
flux_1 = auto() # TODO: Extract flux and create "IMAGE_GENERATION_BASELINE_CATEGORY" due to name inconsistency


MODEL_PURPOSE_LOOKUP: dict[MODEL_REFERENCE_CATEGORY, MODEL_PURPOSE] = {
Expand Down

0 comments on commit 4e01568

Please sign in to comment.