mirror of
https://github.com/PaddlePaddle/FastDeploy.git
synced 2025-11-02 12:44:20 +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:
@@ -46,6 +46,10 @@ class EngineArgs:
|
||||
"""
|
||||
The name or path of the model to be used.
|
||||
"""
|
||||
revision: Optional[str] = "master"
|
||||
"""
|
||||
The revision for downloading models.
|
||||
"""
|
||||
model_config_name: Optional[str] = "config.json"
|
||||
"""
|
||||
The name of the model configuration file.
|
||||
@@ -340,6 +344,12 @@ class EngineArgs:
|
||||
default=EngineArgs.model,
|
||||
help="Model name or path to be used.",
|
||||
)
|
||||
model_group.add_argument(
|
||||
"--revision",
|
||||
type=nullable_str,
|
||||
default=EngineArgs.revision,
|
||||
help="Revision for downloading models",
|
||||
)
|
||||
model_group.add_argument(
|
||||
"--model-config-name",
|
||||
type=nullable_str,
|
||||
|
||||
Reference in New Issue
Block a user