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
+5
-1
@@ -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):
|
||||
|
||||
Reference in new issue
Block a user