Add txt extension to DDUF

#1
by coyotte508 HF staff - opened
Files changed (1) hide show
  1. src/lib/check-filename.ts +3 -1
src/lib/check-filename.ts CHANGED
@@ -3,9 +3,11 @@ export function checkFilename(filename: string) {
3
  !filename.endsWith('.safetensors') &&
4
  !filename.endsWith('.json') &&
5
  !filename.endsWith('.gguf') &&
 
 
6
  !filename.endsWith('/')
7
  ) {
8
- throw new Error('Files must have a .safetensors, .gguf or .json extension');
9
  }
10
 
11
  const split = filename.split('/');
 
3
  !filename.endsWith('.safetensors') &&
4
  !filename.endsWith('.json') &&
5
  !filename.endsWith('.gguf') &&
6
+ !filename.endsWith('.txt') &&
7
+ !filename.endsWith('.model') &&
8
  !filename.endsWith('/')
9
  ) {
10
+ throw new Error('Files must have a .safetensors, .txt, .model, .gguf or .json extension');
11
  }
12
 
13
  const split = filename.split('/');