alphard/ComfyUI-Lora-Manager
1
0
Fork 0
mirror of https://github.com/willmiao/ComfyUI-Lora-Manager.git synced 2026-04-12 05:42:14 -03:00
Code Issues Packages Projects Releases Wiki Activity

fix(usage): resolve checkpoint hashes from disk

Browse Source
This commit is contained in:
Will Miao
2026-04-10 22:28:04 +08:00
parent 39643eb2bc
commit 25fa175aa2
4 changed files with 227 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

78
tests/utils/test_usage_stats.py
View File

@@ -194,6 +194,84 @@ async def test_usage_stats_calculates_pending_checkpoint_hash_on_demand(tmp_path
await _finalize_usage_stats(tasks)
async def test_usage_stats_skips_failed_checkpoint_hash_retry(tmp_path, monkeypatch):
stats, tasks, _ = _prepare_usage_stats(tmp_path, monkeypatch)
metadata_payload = {
"models": {
"1": {"type": "checkpoint", "name": "failed_model.safetensors"},
},
"loras": {},
}
checkpoint_cache = SimpleNamespace(
raw_data=[
{
"file_name": "failed_model",
"model_name": "failed_model",
"file_path": "/models/failed_model.safetensors",
"sha256": "",
"hash_status": "failed",
}
]
)
checkpoint_scanner = SimpleNamespace(
get_hash_by_filename=lambda name: None,
get_cached_data=AsyncMock(return_value=checkpoint_cache),
calculate_hash_for_model=AsyncMock(return_value=None),
)
lora_scanner = SimpleNamespace(get_hash_by_filename=lambda name: None)
monkeypatch.setattr(ServiceRegistry, "get_checkpoint_scanner", AsyncMock(return_value=checkpoint_scanner))
monkeypatch.setattr(ServiceRegistry, "get_lora_scanner", AsyncMock(return_value=lora_scanner))
await stats._process_metadata(metadata_payload)
checkpoint_scanner.calculate_hash_for_model.assert_not_awaited()
assert stats.stats["checkpoints"] == {}
await _finalize_usage_stats(tasks)
async def test_usage_stats_resolves_manually_copied_checkpoint_from_disk(tmp_path, monkeypatch):
stats, tasks, _ = _prepare_usage_stats(tmp_path, monkeypatch)
checkpoints_root = tmp_path / "checkpoints"
checkpoints_root.mkdir()
checkpoint_file = checkpoints_root / "Chroma1-HD-Q8_0.gguf"
checkpoint_file.write_text("manual checkpoint content", encoding="utf-8")
metadata_payload = {
"models": {
"1": {"type": "checkpoint", "name": "Chroma1-HD-Q8_0"},
},
"loras": {},
}
checkpoint_cache = SimpleNamespace(raw_data=[])
checkpoint_scanner = SimpleNamespace(
get_hash_by_filename=lambda name: None,
get_cached_data=AsyncMock(return_value=checkpoint_cache),
get_model_roots=lambda: [str(checkpoints_root)],
file_extensions={".ckpt", ".pt", ".pt2", ".bin", ".pth", ".safetensors", ".pkl", ".sft", ".gguf"},
calculate_hash_for_model=AsyncMock(return_value="resolved-hash"),
)
monkeypatch.setattr(ServiceRegistry, "get_checkpoint_scanner", AsyncMock(return_value=checkpoint_scanner))
monkeypatch.setattr(ServiceRegistry, "get_lora_scanner", AsyncMock(return_value=None))
await stats._process_metadata(metadata_payload)
checkpoint_scanner.calculate_hash_for_model.assert_awaited_once_with(
str(checkpoint_file).replace(os.sep, "/")
)
today = datetime.now().strftime("%Y-%m-%d")
assert stats.stats["checkpoints"]["resolved-hash"]["history"][today] == 1
await _finalize_usage_stats(tasks)
async def test_usage_stats_skips_name_fallback_for_missing_lora_hash(tmp_path, monkeypatch):
stats, tasks, _ = _prepare_usage_stats(tmp_path, monkeypatch)