# Copyright 2020 The HuggingFace Team. All rights reserved. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. # You may obtain a copy of the License at # # http://www.apache.org/licenses/LICENSE-2.0 # # Unless required by applicable law or agreed to in writing, software # distributed under the License is distributed on an "AS IS" BASIS, # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. from huggingface_hub import constants from huggingface_hub.cli._cli_utils import check_cli_update, typer_factory from huggingface_hub.cli.auth import auth_cli from huggingface_hub.cli.cache import cache_cli from huggingface_hub.cli.download import download from huggingface_hub.cli.inference_endpoints import ie_cli from huggingface_hub.cli.jobs import jobs_cli from huggingface_hub.cli.lfs import lfs_enable_largefiles, lfs_multipart_upload from huggingface_hub.cli.repo import repo_cli from huggingface_hub.cli.repo_files import repo_files_cli from huggingface_hub.cli.system import env, version from huggingface_hub.cli.upload import upload from huggingface_hub.cli.upload_large_folder import upload_large_folder from huggingface_hub.utils import logging app = typer_factory(help="Hugging Face Hub CLI") # top level single commands (defined in their respective files) app.command(help="Download files from the Hub.")(download) app.command(help="Upload a file or a folder to the Hub.")(upload) app.command(help="Upload a large folder to the Hub. Recommended for resumable uploads.")(upload_large_folder) app.command(name="env", help="Print information about the environment.")(env) app.command(help="Print information about the hf version.")(version) app.command(help="Configure your repository to enable upload of files > 5GB.", hidden=True)(lfs_enable_largefiles) app.command(help="Upload large files to the Hub.", hidden=True)(lfs_multipart_upload) # command groups app.add_typer(auth_cli, name="auth") app.add_typer(cache_cli, name="cache") app.add_typer(repo_cli, name="repo") app.add_typer(repo_files_cli, name="repo-files") app.add_typer(jobs_cli, name="jobs") app.add_typer(ie_cli, name="endpoints") def main(): if not constants.HF_DEBUG: logging.set_verbosity_info() check_cli_update() app() if __name__ == "__main__": main()