mirror of
https://github.com/huggingface/transformers.git
synced 2025-07-03 21:00:08 +06:00
44b21f117b
* Allow dynamic modules to use relative imports * Work for configs * Fix last merge conflict * Save code of registered custom objects * Map strings to strings * Fix test * Add tokenizer * Rework tests * Tests * Ignore fixtures py files for tests * Tokenizer test + fix collection * With full path * Rework integration * Fix typo * Remove changes in conftest * Test for tokenizers * Add documentation * Update docs/source/custom_models.mdx Co-authored-by: Lysandre Debut <lysandre@huggingface.co> * Add file structure and file content * Add more doc * Style * Update docs/source/custom_models.mdx Co-authored-by: Suraj Patil <surajp815@gmail.com> * Address review comments Co-authored-by: Lysandre Debut <lysandre@huggingface.co> Co-authored-by: Suraj Patil <surajp815@gmail.com>
10 lines
226 B
Python
10 lines
226 B
Python
from transformers import PretrainedConfig
|
|
|
|
|
|
class CustomConfig(PretrainedConfig):
|
|
model_type = "custom"
|
|
|
|
def __init__(self, attribute=1, **kwargs):
|
|
self.attribute = attribute
|
|
super().__init__(**kwargs)
|