dduf-check / src /lib /check-filename.ts
coyotte508's picture
coyotte508 HF Staff
Also allow .model files
b6eff4d verified
raw
history blame
499 Bytes
export function checkFilename(filename: string) {
if (
!filename.endsWith('.safetensors') &&
!filename.endsWith('.json') &&
!filename.endsWith('.gguf') &&
!filename.endsWith('.txt')
!filename.endsWith('.model') &&
!filename.endsWith('/')
) {
throw new Error('Files must have a .safetensors, .txt, .model, .gguf or .json extension');
}
const split = filename.split('/');
if (split.length > 2) {
throw new Error('Files must be only one level deep, not more');
}
}