diff --git a/src/lib/model-props.ts b/src/lib/model-props.ts index f5076ca1..16255950 100644 --- a/src/lib/model-props.ts +++ b/src/lib/model-props.ts @@ -285,7 +285,7 @@ export const MODEL_PROPS: Readonly> = { type: { name: 'Type', type: 'string', - enum: ['pth', 'onnx'], + enum: ['pth', 'onnx', 'safetensors'], }, size: { name: 'Size',