glasses.utils.weights.storage package

Subpackages

• glasses.utils.weights.storage.hubs package

Submodules

glasses.utils.weights.storage.HuggingFaceStorage module

class glasses.utils.weights.storage.HuggingFaceStorage.HuggingFaceStorage(ORGANIZATION: str = 'glasses', root: pathlib.Path = PosixPath('/tmp'))

Bases: glasses.utils.weights.storage.Storage.Storage

ORGANIZATION: str = 'glasses'

get(key: str) → Dict[str, torch.Tensor]

property models: List[str]

put(key: str, model: torch.nn.modules.module.Module)

root: pathlib.Path = PosixPath('/tmp')

glasses.utils.weights.storage.LocalStorage module

class glasses.utils.weights.storage.LocalStorage.LocalStorage(root: pathlib.Path = PosixPath('/tmp/glasses'), override: bool = False, fmt: str = 'pth')

Bases: glasses.utils.weights.storage.Storage.Storage

fmt: str = 'pth'

get(key: str) → Dict[str, torch.Tensor]

property models: List[str]

override: bool = False

put(key: str, model: torch.nn.modules.module.Module)

root: pathlib.Path = PosixPath('/tmp/glasses')

glasses.utils.weights.storage.Storage module

class glasses.utils.weights.storage.Storage.Storage

Bases: abc.ABC

abstract get(key: str, **kwargs: Any) → Any

abstract property models: List[str]

abstract put(*args: Any, **kwargs: Any)

Module contents