diff --git a/horde/apis/models/kobold_v2.py b/horde/apis/models/kobold_v2.py index 39e51833..d6593678 100644 --- a/horde/apis/models/kobold_v2.py +++ b/horde/apis/models/kobold_v2.py @@ -238,9 +238,9 @@ def __init__(self, api): }, ) self.response_model_job_pop = api.model( - "GenerationPayloadKobold", + "GenerationPayload", { - "payload": fields.Nested(self.root_model_generation_payload_kobold, skip_none=True), + "payload": fields.Nested(self.response_model_generation_payload, skip_none=True), "id": fields.String(description="The UUID for this text generation."), "ids": fields.List( fields.String(