| from modeling_simple import SimpleModel, SimpleConfig | |
| config = SimpleConfig( | |
| vocab_size=100, | |
| hidden_size=32, | |
| num_labels=2 | |
| ) | |
| model = SimpleModel(config) | |
| # This creates: | |
| # - config.json | |
| # - pytorch_model.bin | |
| model.save_pretrained("simple-model") | |
| config.save_pretrained("simple-model") | |