A tool that can download Hugging Face model(s)
Project description
This is a tool that can automatically download models from Hugging Face(Windows or Linux).You can download single model or make batch download.
-
Download "roberta-base" from Hugging face to "models"
import saodownload as sd sd.download_models('models', ['gpt2'])
And then, the whole gpt2 model in pytorch will be stored in "models" direct.
-
Batch Download pytorch models from Hugging Face
import saodownload as sd sd.download_models('models', ['roberta-base', 'gpt2', 'bert-base-uncased'])
This code will download roberta-base、 gpt2、 bert-base-uncased into "models"
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
niu_download-0.0.1.tar.gz
(4.1 kB
view hashes)
Built Distribution
Close
Hashes for niu_download-0.0.1-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 982c36a4bf6e2cdc7726f2e96ecfcd0a97132fcba79b148612ca1d580addd0b7 |
|
MD5 | 69b73ae7968f660407b504027dd8ba30 |
|
BLAKE2b-256 | 2ff63169b876629a2d232ddda012591111a72e8b4fb246c4db0462f2693c7563 |