1
0
Fork 0

update schemas for archive converter

This commit is contained in:
Sean Sube 2023-12-16 22:49:29 -06:00
parent 72d0d374e1
commit a65e0fd602
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
2 changed files with 4 additions and 1 deletions

View File

@ -23,6 +23,7 @@ def convert_extract_archive(
if path.exists(dest_path):
logger.info("destination path already exists, skipping extraction")
return False, dest_path
cache_path = fetch_model(conversion, name, model["source"], format=format)
@ -33,6 +34,7 @@ def convert_extract_archive(
logger.debug("archive is valid, extracting all files: %s", names)
zip.extractall(path=dest_path)
return True, dest_path
SAFE_NAME = r"^[-_a-zA-Z/\\\.]+$"

View File

@ -12,7 +12,7 @@ $defs:
tensor_format:
type: string
enum: [bin, ckpt, onnx, pt, pth, safetensors]
enum: [bin, ckpt, onnx, pt, pth, safetensors, zip]
embedding_network:
type: object
@ -111,6 +111,7 @@ $defs:
pipeline:
type: string
enum: [
archive,
controlnet,
img2img,
inpaint,