mirror of
https://github.com/willmiao/ComfyUI-Lora-Manager.git
synced 2026-03-22 05:32:12 -03:00
53c4165d82
- Add MODEL_NAME_PATTERN regex to extract model names from parameters - Extract model hash from parsed hashes when available in metadata - Add checkpoint model hash and name extraction from parameters section - Implement checkpoint resource processing from Civitai metadata - Improve model information completeness for better recipe tracking
62 lines
2.0 KiB
Python
62 lines
2.0 KiB
Python
import pytest
|
|
|
|
from py.recipes.parsers.meta_format import MetaFormatParser
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_meta_format_parser_extracts_checkpoint_from_model_hash(monkeypatch):
|
|
checkpoint_info = {
|
|
"id": 222333,
|
|
"modelId": 999888,
|
|
"model": {"name": "Fluxmania V5P", "type": "checkpoint"},
|
|
"name": "v5p",
|
|
"images": [{"url": "https://image.civitai.com/checkpoints/original=true"}],
|
|
"baseModel": "flux",
|
|
"downloadUrl": "https://civitai.com/api/download/checkpoint",
|
|
"files": [
|
|
{
|
|
"type": "Model",
|
|
"primary": True,
|
|
"sizeKB": 1024,
|
|
"name": "Fluxmania_V5P.safetensors",
|
|
"hashes": {"SHA256": "8AE0583B06"},
|
|
}
|
|
],
|
|
}
|
|
|
|
async def fake_metadata_provider():
|
|
class Provider:
|
|
async def get_model_by_hash(self, model_hash):
|
|
assert model_hash == "8ae0583b06"
|
|
return checkpoint_info, None
|
|
|
|
return Provider()
|
|
|
|
monkeypatch.setattr(
|
|
"py.recipes.parsers.meta_format.get_default_metadata_provider",
|
|
fake_metadata_provider,
|
|
)
|
|
|
|
parser = MetaFormatParser()
|
|
|
|
metadata_text = (
|
|
"Shimmering metal forms\n"
|
|
"Negative prompt: flat color\n"
|
|
"Steps: 25, Sampler: dpmpp_2m_sgm_uniform, Seed: 471889513588087, "
|
|
"Model: Fluxmania V5P.safetensors, Model hash: 8ae0583b06, VAE: ae.sft, "
|
|
"Lora_0 Model name: ArtVador I.safetensors, Lora_0 Model hash: 08f7133a58, "
|
|
"Lora_0 Strength model: 0.65, Lora_0 Strength clip: 0.65"
|
|
)
|
|
|
|
result = await parser.parse_metadata(metadata_text)
|
|
|
|
checkpoint = result.get("checkpoint")
|
|
assert checkpoint is not None
|
|
assert checkpoint["hash"] == "8ae0583b06"
|
|
assert checkpoint["name"] == "Fluxmania V5P"
|
|
assert checkpoint["version"] == "v5p"
|
|
assert checkpoint["file_name"] == "Fluxmania_V5P"
|
|
assert result["model"] == checkpoint
|
|
assert result["base_model"] == "flux"
|
|
assert len(result["loras"]) == 1
|