We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
cli.py
1 parent 50fc27d commit 7a97fd6Copy full SHA for 7a97fd6
server/text_generation_server/cli.py
@@ -180,7 +180,8 @@ def convert_to_safetensors(
180
181
if local_pt_index_file:
182
local_pt_index_file = Path(local_pt_index_file)
183
- local_st_index_file = local_pt_index_file.parent / f"{local_pt_index_file.stem.removeprefix('pytorch_').rstrip('.bin.index')}.safetensors.index.json"
+ st_prefix = local_pt_index_file.stem.removeprefix('pytorch_').rstrip('.bin.index')
184
+ local_st_index_file = local_pt_index_file.parent / f"{st_prefix}.safetensors.index.json"
185
186
if os.path.exists(local_st_index_file):
187
print("Existing .safetensors.index.json file found, remove it first to reconvert")
0 commit comments