andras/great-ai
1
0
Fork 0
Code Issues Pull requests 1 Projects Releases 4 Packages Wiki Activity Actions

Move files

Browse source
This commit is contained in:
Andras Schmelczer 2022-07-10 19:48:22 +02:00
parent b97b20ba88
commit c046d64a98
No known key found for this signature in database
GPG key ID: 0EA1BC97D0AB076E
16 changed files with 35 additions and 36 deletions
Download patch file Download diff file Expand all files Collapse all files

16
great_ai/models/use_model.py
View file

@ -32,17 +32,13 @@ def use_model(
) -> Callable[[F], F]:
"""Inject a model into a function.
Load a model specified by `key` and `version` using the
currently active `LargeFile` implementation. If it's a
single object, it is deserialised using `dill`. If it's
a directory of files, a `pathlib.Path` instance is given.
Load a model specified by `key` and `version` using the currently active `LargeFile`
implementation. If it's a single object, it is deserialised using `dill`. If it's a
directory of files, a `pathlib.Path` instance is given.
By default, the function's `model` parameter is replaced
by the loaded model. This can be customised by changing
`model_kwarg_name`.
Multiple models can be loaded by decorating the same
function with `use_model` multiple times.
By default, the function's `model` parameter is replaced by the loaded model. This
can be customised by changing `model_kwarg_name`. Multiple models can be loaded by
decorating the same function with `use_model` multiple times.
Examples:
>>> from great_ai import save_model