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")