huggingface/text-generation-inference
5
0
Fork 0
mirror of https://github.com/huggingface/text-generation-inference.git synced 2025-09-11 04:14:52 +00:00
Code Issues Packages Projects Releases Wiki Activity

support local model config file

Browse Source
This commit is contained in:
zhangsibo1129 2023-09-26 10:50:24 +08:00
parent c5de7cd886
commit 5f76dae04a
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
server/text_generation_server/cli.py
View File

@ -179,6 +179,9 @@ def download_weights(
import json import json
if is_local_model:
config_filename = os.path.join(model_id, "config.json")
else:
config_filename = hf_hub_download(model_id, revision=revision, filename="config.json") config_filename = hf_hub_download(model_id, revision=revision, filename="config.json")
with open(config_filename, "r") as f: with open(config_filename, "r") as f:
config = json.load(f) config = json.load(f)