trained_models/README.md

Trained Models

Place model folders here, one directory per model. You can populate this directory via the CLI:

mini-transformer-fetch AlaBoussoffara/transformer_test   # tiny demo
mini-transformer-fetch AlaBoussoffara/transformer_small  # larger example

or copy your own exported checkpoints using the layout below:

trained_models/
  my-model/
    configs/
      config_inference.yaml
    checkpoints/
      best.pt
    tokenizer/
      tokenizer.json

The CLI (mini-transformer-infer) and apps will look for configs/config_inference.yaml inside each folder. Relative paths in that config should point to files inside the same model directory.