yobx.torch.in_transformers.models.configs#
Function yobx.torch.in_transformers.models.configs.get_cached_configuration lets the user retrieve configurations coming from transformers. Their purpose is to be able to write unit tests without downloading a configuration. Currently, the module contains the following cached configuration:
<<<
import pprint
from yobx.torch.in_transformers.models._configs import _retrieve_cached_configurations
configs = _retrieve_cached_configurations()
pprint.pprint(sorted(configs))
>>>
['arnir0/Tiny-LLM', 'local/GraniteMoeHybrid', 'openai/whisper-tiny']
[runpythonerror]
/opt/hostedtoolcache/Python/3.12.14/x64/lib/python3.12/site-packages/torch/jit/_script.py:1491: FutureWarning: `torch.jit.script` is deprecated. Please switch to `torch.compile` or `torch.export`.
warnings.warn(
To get them, yobx.torch.in_transformers.models.configs.get_cached_configuration("<name>", **kwargs)
where kwargs can be used to overwrite some values.