transformers_config

Classes

TransformersConfig

Configuration class for transformers.

Module Contents

class transformers_config.TransformersConfig(model, tokenizer, vocab_size=None, device='cuda', *args, **kwargs)

Configuration class for transformers.

device

model

tokenizer

vocab_size

gen_kwargs