Centralize image family runtime and workflow adapters
This commit is contained in:
1 parent
7826388cde
commit
b5ee7c1e01
7 files changed
+51
-19
No files matched your search
@@ -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()
|
||||
|
||||
Reference in new issue
Block a user