[Feature] multi-source download (#2986)

* multi-source download

* multi-source download

* huggingface download revision

* requirement

* style

* add revision arg

* test

* pre-commit
This commit is contained in:
Yzc216
2025-07-24 14:26:37 +08:00
committed by GitHub
parent 87a2f4191d
commit e14587a954
7 changed files with 106 additions and 15 deletions

View File

@@ -66,10 +66,11 @@ class LLM:
def __init__(
self,
model: str,
revision: Optional[str] = "master",
tokenizer: Optional[str] = None,
**kwargs,
):
model = retrive_model_from_server(model)
model = retrive_model_from_server(model, revision)
engine_args = EngineArgs(
model=model,
tokenizer=tokenizer,