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 for pushing to Hub.
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 ready to push to Hub |
main(cfg)
Main function for Push to Hub.