This commit is contained in:
Lincoln Stein
2023-11-12 14:20:32 -05:00
parent 024a156114
commit ef8dcf5fae
2 changed files with 8 additions and 7 deletions

View File

@ -49,7 +49,8 @@ def test_type(store: ModelRecordServiceBase):
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",