Skip to main content

HuggingFace model downloader

Project description

hf-model-downloader

Install

pip install hf-model-downloader

pip install git+https://github.com/muellerzr/hf-model-downloader

What is it?

An easy way to download models quickly from huggingface.co with specified backends to specific directories.

While on its own you can pass in --local_dir when using the huggingface_hub CLI, many libraries (such as all of the meta-llama models) come with two sets of weights in the repo, original and safetensor varients in the same branch. As a result, this can bloat our downloads when we don't need them.

hf-downloader solves this by making it easier to download weights to wherever you want, specifying the weight revision you need (and then will also download the relevant config files, etc)

As this is a wrapper around the huggingface_hub CLI, tokens can be passed in via --token or by using huggingface-cli login (recommended)

Examples with meta-llama/Meta-Llama-3.1-8B-Instruct

Downloading the .safetensors version (default)

download-model --backend safetensors meta-llama/Meta-Llama-3.1-8B-Instruct 

Downloading the .pth version

hf-downloader --backend torch meta-llama/Meta-Llama-3.1-8B-Instruct 

Downloading to a specific directory

hf-downloader --local-dir my_llama meta-llama/Meta-Llama-3.1-8B-Instruct 

Using hf_transfer for faster downloads

hf-downloader --fast meta-llama/Meta-Llama-3.1-8B-Instruct

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

hf-model-downloader-0.0.1.tar.gz (11.3 kB view hashes)

Uploaded Source

Built Distribution

hf_model_downloader-0.0.1-py3-none-any.whl (11.8 kB view hashes)

Uploaded Python 3

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page