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:
Lincoln Stein
2024-04-24 23:53:30 -04:00
parent 34cdfc61ab
commit d72f272f16
7 changed files with 64 additions and 123 deletions

View File

@ -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