azula.hub¶

Utilities for downloading models.

Functions¶

`get_dir`	Returns the cache directory used for storing models & weights.
`set_dir`	Sets the cache directory used for storing models & weights.
`download`	Downloads data at a given URL to a local file.

azula.hub.get_dir()¶

Returns the cache directory used for storing models & weights.

azula.hub.set_dir(cache_dir)¶

Sets the cache directory used for storing models & weights.

azula.hub.download(url, filename=None, hash_prefix=None, quiet=False)¶

Downloads data at a given URL to a local file.

Parameters:

url (str) – A URL. Google Drive URLs are supported.
filename (str | None) – A local file name. If None, use the sanitized URL instead. If a file with the same name exists, the download is skipped.
hash_prefix (str | None) – The expected hash prefix of the file, formatted as "alg:prefix".
quiet (bool) – Whether to keep it quiet in the terminal or not.