mirror of
https://github.com/PaddlePaddle/FastDeploy.git
synced 2025-10-05 08:37:06 +08:00
[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:
@@ -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,
|
||||
|
Reference in New Issue
Block a user