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

Wrap in try except for InvalidModelConfigException

Browse Source
This commit is contained in:
Brandon Rising 2024-03-18 12:10:06 -04:00
parent c778d74a42
commit 1f000306f3
1 changed files with 15 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tests/app/services/model_install/test_model_install.py
View File

@ -298,6 +298,7 @@ def test_heuristic_import_with_type(mm2_installer: ModelInstallServiceBase, mode
"hash": "placeholder2", "hash": "placeholder2",
} }
assert "repo_id" in model_params assert "repo_id" in model_params
try:
install_job1 = mm2_installer.heuristic_import(source=model_params["repo_id"], config=config1) install_job1 = mm2_installer.heuristic_import(source=model_params["repo_id"], config=config1)
mm2_installer.wait_for_installs(timeout=20) mm2_installer.wait_for_installs(timeout=20)
if model_params["type"] != "embedding": if model_params["type"] != "embedding":
@ -311,3 +312,5 @@ def test_heuristic_import_with_type(mm2_installer: ModelInstallServiceBase, mode
mm2_installer.wait_for_installs(timeout=20) mm2_installer.wait_for_installs(timeout=20)
assert install_job2.complete assert install_job2.complete
assert install_job2.config_out if model_params["type"] == "embedding" else not install_job2.config_out assert install_job2.config_out if model_params["type"] == "embedding" else not install_job2.config_out
except InvalidModelConfigException:
assert model_params["type"] != "embedding"