Skip to content

xlm.commands.push_to_hub

PushToHubConfig dataclass

Configuration for pushing model to Hugging Face Hub.

instantiate_model(cfg, datamodule, tokenizer)

Instantiate a model from checkpoint or config.

Parameters:

Name Type Description Default
cfg DictConfig

Hydra config

required
datamodule Any

Datamodule instance

required
tokenizer Any

Tokenizer instance

required

Returns:

Name Type Description
Harness Harness

The instantiated model

main(cfg)

Main function for Push to Hub.