214 lines
4.3 KiB
YAML
214 lines
4.3 KiB
YAML
$id: https://github.com/ssube/onnx-web/blob/main/api/schemas/extras.yaml
|
|
$schema: https://json-schema.org/draft/2020-12/schema
|
|
|
|
$defs:
|
|
legacy_tuple:
|
|
type: array
|
|
items:
|
|
oneOf:
|
|
- type: boolean
|
|
- type: number
|
|
- type: string
|
|
|
|
tensor_format:
|
|
type: string
|
|
enum: [bin, ckpt, onnx, pt, pth, safetensors, zip]
|
|
|
|
embedding_network:
|
|
type: object
|
|
required: [name, source]
|
|
properties:
|
|
format:
|
|
$ref: "#/$defs/tensor_format"
|
|
label:
|
|
type: string
|
|
model:
|
|
type: string
|
|
enum: [concept, embeddings]
|
|
name:
|
|
type: string
|
|
source:
|
|
type: string
|
|
token:
|
|
type: string
|
|
type:
|
|
type: string
|
|
const: inversion # TODO: add embedding
|
|
weight:
|
|
type: number
|
|
|
|
lora_network:
|
|
type: object
|
|
required: [name, source, type]
|
|
properties:
|
|
format:
|
|
$ref: "#/$defs/tensor_format"
|
|
label:
|
|
type: string
|
|
model:
|
|
type: string
|
|
enum: [cloneofsimo, sd-scripts]
|
|
name:
|
|
type: string
|
|
source:
|
|
type: string
|
|
tokens:
|
|
type: array
|
|
items:
|
|
type: string
|
|
type:
|
|
type: string
|
|
const: lora
|
|
weight:
|
|
type: number
|
|
|
|
base_model:
|
|
type: object
|
|
required: [name, source]
|
|
properties:
|
|
format:
|
|
$ref: "#/$defs/tensor_format"
|
|
half:
|
|
type: boolean
|
|
label:
|
|
type: string
|
|
name:
|
|
type: string
|
|
opset:
|
|
type: number
|
|
source:
|
|
type: string
|
|
|
|
correction_model:
|
|
allOf:
|
|
- $ref: "#/$defs/base_model"
|
|
- type: object
|
|
properties:
|
|
model:
|
|
type: string
|
|
enum: [
|
|
codeformer,
|
|
gfpgan
|
|
]
|
|
|
|
diffusion_model:
|
|
allOf:
|
|
- $ref: "#/$defs/base_model"
|
|
- type: object
|
|
properties:
|
|
config:
|
|
type: string
|
|
hash:
|
|
type: string
|
|
image_size:
|
|
type: number
|
|
inversions:
|
|
type: array
|
|
items:
|
|
$ref: "#/$defs/embedding_network"
|
|
loras:
|
|
type: array
|
|
items:
|
|
$ref: "#/$defs/lora_network"
|
|
pipeline:
|
|
type: string
|
|
enum: [
|
|
archive,
|
|
controlnet,
|
|
img2img,
|
|
inpaint,
|
|
lpw,
|
|
panorama,
|
|
pix2pix,
|
|
txt2img,
|
|
txt2img-optimum,
|
|
txt2img-sdxl,
|
|
upscale,
|
|
]
|
|
vae:
|
|
type: string
|
|
version:
|
|
type: string
|
|
enum: [
|
|
v1,
|
|
v2,
|
|
v2.1,
|
|
]
|
|
|
|
upscaling_model:
|
|
allOf:
|
|
- $ref: "#/$defs/base_model"
|
|
- type: object
|
|
required: [scale]
|
|
properties:
|
|
model:
|
|
type: string
|
|
enum: [
|
|
bsrgan,
|
|
resrgan,
|
|
swinir
|
|
]
|
|
scale:
|
|
type: number
|
|
|
|
source_model:
|
|
type: object
|
|
required: [name, source]
|
|
properties:
|
|
dest:
|
|
type: string
|
|
format:
|
|
type: string
|
|
name:
|
|
type: string
|
|
source:
|
|
type: string
|
|
|
|
translation:
|
|
type: object
|
|
additionalProperties: False
|
|
patternProperties:
|
|
"(\\w|\\.|-)+":
|
|
oneOf:
|
|
- type: string
|
|
- $ref: "#/$defs/translation"
|
|
|
|
type: object
|
|
additionalProperties: False
|
|
properties:
|
|
diffusion:
|
|
type: array
|
|
items:
|
|
oneOf:
|
|
- $ref: "#/$defs/legacy_tuple"
|
|
- $ref: "#/$defs/diffusion_model"
|
|
correction:
|
|
type: array
|
|
items:
|
|
oneOf:
|
|
- $ref: "#/$defs/legacy_tuple"
|
|
- $ref: "#/$defs/correction_model"
|
|
networks:
|
|
type: array
|
|
items:
|
|
oneOf:
|
|
- $ref: "#/$defs/lora_network"
|
|
- $ref: "#/$defs/embedding_network"
|
|
sources:
|
|
type: array
|
|
items:
|
|
oneOf:
|
|
- $ref: "#/$defs/legacy_tuple"
|
|
- $ref: "#/$defs/source_model"
|
|
strings:
|
|
type: object
|
|
additionalProperties: False
|
|
patternProperties:
|
|
"^\\w\\w$":
|
|
$ref: "#/$defs/translation"
|
|
upscaling:
|
|
type: array
|
|
items:
|
|
oneOf:
|
|
- $ref: "#/$defs/legacy_tuple"
|
|
- $ref: "#/$defs/upscaling_model" |