Centralize image family runtime and workflow adapters

This commit is contained in:
Mikei386 committed 2026-10-01 14:13:47 +02:00
1 parent 7826388cde
commit b5ee7c1e01
7 files changed
+51 -19

No files matched your search

+12
View File
@@ -40,6 +40,18 @@ class ImageTestTests(unittest.TestCase):
self.assertEqual(image_capabilities(dict(repo=FLUX_REPO,file='Flux.2 Klein-9B_fp16_nsfw.safetensors'))['reference_images'],4)
self.assertEqual(image_capabilities(dict(repo=QWEN_REPO,file='model.gguf'))['reference_images'],4)
self.assertEqual(image_capabilities(dict(repo='unknown',file='model.safetensors'))['reference_images'],0)
def test_family_registry_reuses_workflow_from_lineage_not_filename(self):
from profiles import image_adapter,image_recipe,FLUX_COMPONENTS
m=dict(repo='someone/new-weights',file='custom.safetensors',runtime_metadata={'base_models':['black-forest-labs/FLUX.2-klein-9B']})
self.assertEqual(image_adapter(m)['workflow'],'flux')
self.assertEqual(image_recipe(m),FLUX_COMPONENTS)
m['runtime_metadata']={};m['file']='flux-qwen-looking-name.safetensors'
self.assertIsNone(image_adapter(m))
m['runtime_metadata']={'architecture':'Flux2Transformer2DModel'}
self.assertIsNone(image_adapter(m))
m['runtime_metadata']={'architecture':'qwen_image_2_1'};m['file']='new.gguf'
self.assertEqual(image_adapter(m)['workflow'],'qwen')
m['file']='vae/new.gguf';self.assertIsNone(image_adapter(m))
def test_reference_validation_enforces_profile_limit_and_signature(self):
import base64
png='data:image/png;base64,'+base64.b64encode(b'\x89PNG\r\n\x1a\nsynthetic').decode()