piscis.downloads

Functions

download_pretrained_model(→ None)	Download a pretrained model from Hugging Face.
list_pretrained_models(→ Sequence[str])	List pretrained models from Hugging Face.
download_dataset(→ None)	Download a dataset from Hugging Face.
list_datasets(→ Sequence[str])	List datasets from Hugging Face.

Module Contents

piscis.downloads.download_pretrained_model(model_name: str) → None

Download a pretrained model from Hugging Face.

Parameters:

model_namestr

Model name.

Raises:

ValueError

If model_name is not found.

piscis.downloads.list_pretrained_models() → Sequence[str]

List pretrained models from Hugging Face.

Returns:

pretrained_modelsSequence[str]

List of pretrained model names.

piscis.downloads.download_dataset(dataset_name: str, path: str, minimal_download: bool = True) → None

Download a dataset from Hugging Face.

Parameters:

dataset_namestr

Dataset name.

pathstr

Path to save the dataset.

Raises:

ValueError

If dataset_name is not found.

piscis.downloads.list_datasets() → Sequence[str]

List datasets from Hugging Face.

Returns:

datasetsSequence[str]

List of dataset names.