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

+5 -1
View File
@@ -89,9 +89,13 @@ def runtime_metadata(model):
architecture=next((x for x in architectures if isinstance(x,str) and 0<len(x)<=100),None)
architecture=architecture or next((config.get(k) for k in ('_class_name','model_type') if isinstance(config.get(k),str) and 0<len(config[k])<=100),None)
tags=[x for x in model.get('tags',[]) if isinstance(x,str) and len(x)<=80][:32] if isinstance(model.get('tags'),list) else []
card=model.get('cardData') if isinstance(model.get('cardData'),dict) else {}
base=card.get('base_model',[])
base=[base] if isinstance(base,str) else base if isinstance(base,list) else []
base_models=[x for x in base if isinstance(x,str) and re.fullmatch(r'[A-Za-z0-9_.-]+/[A-Za-z0-9_.-]+',x)][:8]
return dict(pipeline_tag=model.get('pipeline_tag') if isinstance(model.get('pipeline_tag'),str) and len(model['pipeline_tag'])<=80 else None,
library_name=model.get('library_name') if isinstance(model.get('library_name'),str) and len(model['library_name'])<=80 else None,
architecture=architecture,tags=tags)
architecture=architecture,tags=tags,base_models=base_models)
class Catalog:
def __init__(self,root):