mirror of
https://github.com/invoke-ai/InvokeAI
synced 2024-08-30 20:32:17 +00:00
Address change requests in first round of PR reviews.
Pending: - Move model install calls into model manager and create passthrus in invocation_context. - Consider splitting load_model_from_url() into a call to get the path and a call to load the path.
This commit is contained in:
@ -49,9 +49,3 @@ def test_download_and_load(mock_context: InvocationContext):
|
||||
assert isinstance(model_1, dict)
|
||||
|
||||
|
||||
def test_install_model(mock_context: InvocationContext):
|
||||
key = mock_context.models.install_model("https://www.test.foo/download/test_embedding.safetensors")
|
||||
assert key is not None
|
||||
model = mock_context.models.load(key)
|
||||
assert model is not None
|
||||
assert model.config.key == key
|
||||
|
Reference in New Issue
Block a user