update schemas for archive converter

ssube · Dec 17, 2023 · a65e0fd · a65e0fd
1 parent 72d0d37
commit a65e0fd
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/api/onnx_web/convert/archive.py b/api/onnx_web/convert/archive.py
@@ -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/\\\.]+$"

diff --git a/api/schemas/extras.yaml b/api/schemas/extras.yaml
@@ -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,