Skip to content

Commit 3e34359

Browse files
committed
Add .model extension to default "extra" files downloaded with weights
When using the download-weights CLI command and specifying a single extension. This is used for slow tokenizers which can be subsequently converted to fast tokenizers.
1 parent 1130840 commit 3e34359

File tree

1 file changed

+1
-1
lines changed
  • server/text_generation_server

1 file changed

+1
-1
lines changed

server/text_generation_server/cli.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -66,7 +66,7 @@ def download_weights(
6666
):
6767
from text_generation_server import utils
6868

69-
meta_exts = [".json", ".py", ".md"]
69+
meta_exts = [".json", ".py", ".model", ".md"]
7070

7171
extensions = extension.split(",")
7272

0 commit comments

Comments
 (0)