File size: 353 Bytes
3b2d368 | 1 2 3 4 5 6 7 8 9 10 | from hydra.utils import get_class
from lmr.tokenizer import Tokenizer
def get_model(model_config, vocab_size):
model_config["vocab_size"] = vocab_size
model = get_class(model_config.model_class)(model_config)
model.name = model_config.model_name
model.full_name = f"{model_config.model_name}_{model_config.size_name}"
return model |