Revert "fix: use the correct response model for text async (#410)"

This reverts commit 0156cdd.
Haidra-Org · May 21, 2024 · 64a9c47 · 64a9c47
1 parent 0d790f1
commit 64a9c47
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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(