File size: 300 Bytes
2216aae | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 | 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")
|