diff --git a/horde_sdk/ai_horde_api/apimodels/generate/_async.py b/horde_sdk/ai_horde_api/apimodels/generate/_async.py index 3ec5db8..f69838e 100644 --- a/horde_sdk/ai_horde_api/apimodels/generate/_async.py +++ b/horde_sdk/ai_horde_api/apimodels/generate/_async.py @@ -1,4 +1,4 @@ -from pydantic import model_validator +from pydantic import AliasChoices, Field, model_validator from typing_extensions import override from horde_sdk.ai_horde_api.apimodels.base import ( @@ -77,6 +77,12 @@ class ImageGenerationInputPayload(HordeAPIObject, ImageGenerateParamMixin): v2 API Model: `ModelGenerationInputStable` """ + steps: int = Field(default=25, ge=1, validation_alias=AliasChoices("steps", "ddim_steps")) + """The number of image generation steps to perform.""" + + n: int = Field(default=1, ge=1, le=20, validation_alias=AliasChoices("n", "n_iter")) + """The number of images to generate. Defaults to 1, maximum is 20.""" + @override @classmethod def get_api_model_name(cls) -> str | None: diff --git a/horde_sdk/ai_horde_api/apimodels/generate/_pop.py b/horde_sdk/ai_horde_api/apimodels/generate/_pop.py index 7fd309e..fcfde51 100644 --- a/horde_sdk/ai_horde_api/apimodels/generate/_pop.py +++ b/horde_sdk/ai_horde_api/apimodels/generate/_pop.py @@ -1,5 +1,5 @@ import pydantic -from pydantic import Field, field_validator +from pydantic import AliasChoices, Field, field_validator from typing_extensions import override from horde_sdk.ai_horde_api.apimodels.base import ( @@ -63,6 +63,11 @@ class ImageGenerateJobPopSkippedStatus(pydantic.BaseModel): class ImageGenerateJobPopPayload(ImageGenerateParamMixin): prompt: str + ddim_steps: int = Field(default=25, ge=1, validation_alias=AliasChoices("steps", "ddim_steps")) + """The number of image generation steps to perform.""" + + n_iter: int = Field(default=1, ge=1, le=20, validation_alias=AliasChoices("n", "n_iter")) + """The number of images to generate. Defaults to 1, maximum is 20.""" class ImageGenerateJobResponse(HordeResponseBaseModel, JobResponseMixin, ResponseRequiringFollowUpMixin): diff --git a/pyproject.toml b/pyproject.toml index bc9b1e0..9cdef93 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta" [project] name = "horde_sdk" -version = "0.7.3" +version = "0.7.4" description = "A python toolkit for interacting with the horde APIs, services, and ecosystem." authors = [ {name = "tazlin", email = "tazlin.on.github@gmail.com"}, @@ -18,7 +18,7 @@ classifiers = [ "Programming Language :: Python :: 3", "Operating System :: OS Independent", "License :: OSI Approved :: GNU Affero General Public License v3", - "Development Status :: 2 - Pre-Alpha", + "Development Status :: 4 - Beta", ] [tool.setuptools.package-dir]