feat: support for lora is_version

Haidra-Org · Dec 26, 2023 · ac672cc · ac672cc
1 parent 7d4ea49
commit ac672cc
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 2 deletions.
diff --git a/horde_sdk/ai_horde_api/apimodels/base.py b/horde_sdk/ai_horde_api/apimodels/base.py
@@ -74,6 +74,8 @@ class LorasPayloadEntry(BaseModel):
     """The strength of the LoRa against the clip model."""
     inject_trigger: str | None = Field(default=None, min_length=1, max_length=30)
     """Any trigger required to activate the LoRa model."""
+    is_version: bool = Field(default=False)
+    """If true, will treat the lora name as a version ID."""
 
 
 class TIPayloadEntry(BaseModel):

diff --git a/tests/test_data/ai_horde_api/example_payloads/_v2_generate_async_post.json b/tests/test_data/ai_horde_api/example_payloads/_v2_generate_async_post.json
@@ -24,7 +24,8 @@
                 "name": "GlowingRunesAIV6",
                 "model": 1.0,
                 "clip": 1.0,
-                "inject_trigger": "a"
+                "inject_trigger": "a",
+                "is_version": false
             }
         ],
         "tis": [

diff --git a/tests/test_data/ai_horde_api/example_responses/_v2_generate_pop_post_200.json b/tests/test_data/ai_horde_api/example_responses/_v2_generate_pop_post_200.json
@@ -23,7 +23,8 @@
                 "name": "GlowingRunesAIV6",
                 "model": 1.0,
                 "clip": 1.0,
-                "inject_trigger": "a"
+                "inject_trigger": "a",
+                "is_version": false
             }
         ],
         "tis": [