register_tokenizer - Tinker Documentation

tinker_cookbook.tokenizer_utils.register_tokenizer

tinker_cookbook.tokenizer_utils.register_tokenizer( name, factory)

Register a custom tokenizer factory.

Once registered, get_tokenizer will call factory() instead of loading from HuggingFace when the given name is requested.

Parameters:

Returns: None

def my_tokenizer_factory():
    return MyCustomTokenizer()
register_tokenizer("Foo/foo_tokenizer", my_tokenizer_factory)

Referenced by