Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

implement workaround for FastAPI and discriminated unions in Body parameter

Browse Source
This commit is contained in:
Lincoln Stein
2023-11-11 12:22:38 -05:00
parent 2b36565e9e
commit af2264b6eb
3 changed files with 46 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
tests/app/services/model_records/test_model_records_sql.py
View File

@ -44,6 +44,12 @@ def example_config() -> TextualInversionConfig:
)
def test_type(store: ModelRecordServiceBase):
config = example_config()
store.add_model("key1", config)
config1 = store.get_model("key1")
assert type(config1) == TextualInversionConfig
def test_add(store: ModelRecordServiceBase):
raw = dict(
path="/tmp/foo.ckpt",