DiffusionAi/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI.git synced 2025-04-04 22:43:40 +08:00
Code Issues Packages Projects Releases Wiki Activity

Skip hashing to make test quicker

Browse Source
This commit is contained in:
Billy 2025-03-18 09:50:18 +11:00
parent 3469fc9843
commit 1953f3cbcd
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
tests/test_model_probe.py
View File

@ -146,18 +146,19 @@ def test_regression_against_model_probe(datadir: Path, override_model_loading):
The test paths are gathered from the 'test_model_probe' directory.
"""
configs_with_tests = set()
model_paths = ModelSearch().search(datadir / "stripped_models")
fake_hash = "abcdefgh" # skip hashing to make test quicker
model_paths = ModelSearch().search(datadir)
for path in model_paths:
legacy_config = new_config = None
try:
legacy_config = ModelProbe.probe(path)
legacy_config = ModelProbe.probe(path, { "hash": fake_hash })
except InvalidModelConfigException:
pass
try:
new_config = ModelConfigBase.classify(path)
new_config = ModelConfigBase.classify(path, hash=fake_hash)
except InvalidModelConfigException:
pass