Skip to main content

Toolkit for managing huggingface models and datasets

Project description

huggingface_tool

Tools for loading, upload, managing huggingface models and datasets

Installation

pip install huggingface_tool

Usage

  • Download and save tokenizer with: htool save-tk <tokenizer_name> <save_dir>
    • For example: htool save-tk gpt2 ./gpt2
  • Download and save dataset with: htool save-data <dataset_name> <save_dir>
    • For example: htool save-data daily_dialog ./daily_dialog
  • Download and save diffusion models with: htool save-dm <model_name> <save_dir>
    • For example: htool save-dm google/ddpm-cat-256 ./google/

Citing huggingface_tool

If our work has been helpful to you, please feel free to cite us:

@misc{huggingface_tool2023,
    title={huggingface_tool},
    author={OpenRL Contributors},
    publisher = {GitHub},
    howpublished = {\url{https://github.com/OpenRL-Lab/huggingface_tool}},
    year={2023},
}

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

huggingface_tool-0.0.1.tar.gz (10.1 kB view hashes)

Uploaded Source

Built Distribution

huggingface_tool-0.0.1-py3-none-any.whl (16.0 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