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

Optional base_name_or_model_path.

Browse Source
This commit is contained in:
Nicolas Patry 2024-05-17 14:20:58 +00:00
parent e5416274df
commit 52c9ff9aca
1 changed files with 30 additions and 28 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
server/text_generation_server/cli.py
View File

@ -171,12 +171,13 @@ def download_weights(
with open(config, "r") as f: with open(config, "r") as f:
config = json.load(f) config = json.load(f)
model_id = config["base_model_name_or_path"] base_model_id = config.get("base_model_name_or_path", None)
if base_model_id:
revision = "main" revision = "main"
try: try:
utils.weight_files(model_id, revision, extension) utils.weight_files(base_model_id, revision, extension)
logger.info( logger.info(
f"Files for parent {model_id} are already present on the host. " f"Files for parent {base_model_id} are already present on the host. "
"Skipping download." "Skipping download."
) )
return return
@ -222,12 +223,13 @@ def download_weights(
with open(config, "r") as f: with open(config, "r") as f:
config = json.load(f) config = json.load(f)
model_id = config["base_model_name_or_path"] base_model_id = config.get("base_model_name_or_path", None)
if base_model_id:
revision = "main" revision = "main"
try: try:
utils.weight_files(model_id, revision, extension) utils.weight_files(base_model_id, revision, extension)
logger.info( logger.info(
f"Files for parent {model_id} are already present on the host. " f"Files for parent {base_model_id} are already present on the host. "
"Skipping download." "Skipping download."
) )
return return