Skip to content

Commit

Permalink
Updated pr.
Browse files Browse the repository at this point in the history
  • Loading branch information
lu-ohai committed Nov 12, 2024
1 parent 212ec33 commit 8d18002
Showing 1 changed file with 10 additions and 0 deletions.
10 changes: 10 additions & 0 deletions ads/aqua/model/model.py
Original file line number Diff line number Diff line change
Expand Up @@ -1287,7 +1287,11 @@ def _download_model_from_hf(
os_path (str): The OS path where the model files are located.
local_dir (str): The local temp dir to store the huggingface model.
allow_patterns (list): Model files matching at least one pattern are downloaded.
Example: ["*.json"] will download all .json files. ["folder/*"] will download all files under `folder`.
Patterns are Standard Wildcards (globbing patterns) and rules can be found here: https://docs.python.org/3/library/fnmatch.html
ignore_patterns (list): Model files matching any of the patterns are not downloaded.
Example: ["*.json"] will ignore all .json files. ["folder/*"] will ignore all files under `folder`.
Patterns are Standard Wildcards (globbing patterns) and rules can be found here: https://docs.python.org/3/library/fnmatch.html
Returns
-------
Expand Down Expand Up @@ -1329,6 +1333,12 @@ def register(
os_path (str): Object storage destination URI to store the downloaded model. Format: oci://bucket-name@namespace/prefix
inference_container (str): selects service defaults
finetuning_container (str): selects service defaults
allow_patterns (list): Model files matching at least one pattern are downloaded.
Example: ["*.json"] will download all .json files. ["folder/*"] will download all files under `folder`.
Patterns are Standard Wildcards (globbing patterns) and rules can be found here: https://docs.python.org/3/library/fnmatch.html
ignore_patterns (list): Model files matching any of the patterns are not downloaded.
Example: ["*.json"] will ignore all .json files. ["folder/*"] will ignore all files under `folder`.
Patterns are Standard Wildcards (globbing patterns) and rules can be found here: https://docs.python.org/3/library/fnmatch.html
Returns:
AquaModel:
Expand Down

0 comments on commit 8d18002

Please sign in to comment.