> When writing text
> with a compatible markup language,
> add a line break
> after each substantial unit of thought.
SemBr is a command-line tool powered by Transformer models that performs semantic linebreaks to breaks lines in a text file at semantic boundaries. It supports multiple file types including LaTeX, Markdown, and plain text, with automatic file type detection.
For coding agents, as an alternative, I also use SemBr.Skill to perform semantic linebreaks and check consistency before and after edits.
SemBr is available as a Python package on PyPI. For the default install:
pip install sembrWith uv:
uv tool install sembrFor CUDA on Linux, install the CUDA extra:
uv tool install "sembr[cuda]"To install the latest development version directly from GitHub:
# Install from GitHub main branch
uv tool install git+https://github.com/admk/sembr.git
# Run directly without installing
uvx --from git+https://github.com/admk/sembr.git sembrNote that the development version may include experimental features and could be less stable than the PyPI release.
To develop this project, clone and install in development mode:
git clone https://github.com/admk/sembr.git
cd sembr
SEMBR_VERSION_SUFFIX=.dev0 \
uv tool install --editable . --force --refresh-package sembrSemBr is supported on Linux, macOS and Windows (well-tested on macOS). On machines with CUDA devices, or on Apple Silicon Macs, SemBr will use the GPU / Apple Neural Engine to accelerate inference.
To use SemBr, run the following command in your terminal:
sembr -i <input_file> -o <output_file>where <input_file> and <output_file>
are the paths to the input and output files respectively.
On the first run,
it will download the SemBr model
and cache it in ~/.cache/huggingface.
Subsequent runs will check for updates
and use the cached model if it is up-to-date.
Alternatively,
you can pipe the input into sembr,
and the output can also be printed to the terminal:
cat <input_file> | sembrThis is especially useful if you want to use SemBr with clipboard managers, for instance, on a Mac:
pbpaste | sembr | pbcopyOr on Linux:
xclip -o | sembr | xclip -iAdditionally,
you can configure SemBr by creating
$XDG_CONFIG_HOME/sembr/config.toml.
If XDG_CONFIG_HOME is not set,
SemBr reads ~/.config/sembr/config.toml.
The complete commented defaults
are stored in sembr/default.toml.
Copy that file to your config path
and edit only the values you want to change.
To use it offline,
you can download the model from Hugging Face
and set model.name to the model directory,
or prepend TRANSFORMERS_OFFLINE=1 to the command
to use the cached model.
You can override config values for a single run
with -c or --config:
sembr \
-c model.name=/path/to/model \
-c optimize.algorithm=balanced_linebreaks \
-c optimize.preferred_min_tokens_per_line=8 \
-c optimize.preferred_max_tokens_per_line=10 \
-c optimize.line_length_penalty_weight=0.05The supported config keys are:
model.name: The name of the Hugging Face model to use.model.bits: Quantization bits for model weights (4or8). Requires CUDA. Not supported on MPS.model.dtype: Data type for model weights (e.g.float16,bfloat16). Default isfloat32.inference.batch_size: The number of lines to process in a batch. Default is8.inference.overlap_divisor: The overlap divisor for tiled inference. Default is8.optimize.algorithm: The prediction function to use. Options areargmax,logit_adjustment,greedy_linebreaks, andbalanced_linebreaks. Default isbalanced_linebreaks.optimize.preferred_min_tokens_per_line: Preferred lower line length target. Default is8.optimize.preferred_max_tokens_per_line: Preferred upper line length target. Default is10.optimize.line_length_penalty_weight: Penalty weight for line lengths outside the preferred range. The default is0.05.format.num_spaces: Number of spaces represented by one indentation level, orautoto detect2,4, or8from the input. The default isauto.format.indent_type: Indentation unit to emit. Options arespace,tab, andauto.autodetects space or tab indentation from the input. The default isspace.listen.host: The host address of the SemBr API server. The default is127.0.0.1.listen.port: The port for the SemBr API server. The default is8384.
You can also specify the following command-line options:
-l,--listen: Serves the SemBr API on a local server.- Each instance of
sembrrun will detect if the API is accessible, and if not it will run the model on its own. - This option is useful to avoid the time taken to initialize the model by keeping it in memory in a separate process.
- Each instance of
--file-type <type>: File type (plaintext,latex,markdown, etc.). Auto-detected using Magika if not provided.--mcp: Start MCP server mode instead of processing text.
The balanced_linebreaks algorithm
optimizes line breaks with dynamic programming
over each parsed paragraph.
It precomputes token costs
from the model log probabilities.
A no-break token costs -log P(off),
and a break token costs -log P(breaks).
After choosing break positions,
it uses the highest-scoring indent level
at each chosen position to recover the break type.
The objective also adds a quadratic penalty
when the token count falls outside
optimize.preferred_min_tokens_per_line
and optimize.preferred_max_tokens_per_line.
Larger optimize.line_length_penalty_weight values
make the algorithm favor the preferred range more strongly.
For a paragraph with n tokens,
the implementation uses prefix sums,
a monotonic queue for the no-penalty range,
and a Li Chao tree for long-line penalties.
The optimization complexity is
O(n * l + n log n),
where l is optimize.preferred_min_tokens_per_line.
Memory usage is O(n) per paragraph.
Alternatively,
you can run sembr as an MCP server.
Simply add the following configuration
to your MCP server configuration:
"mcpServers": {
"sembr": {
"type": "stdio",
"command": "uvx",
"args": [
"sembr",
"--mcp"
],
}
}The server also supports the formatting options described above.
It will expose a wrap_text tool
for the MCP client to use.
Semantic Line Breaks or Semantic Linefeeds describe a set of conventions for using insensitive vertical whitespace to structure prose along semantic boundaries.
Semantic Line Breaks has the following advantages:
-
Breaking lines by splitting clauses reflects the logical, grammatical and semantic structure of the text.
-
It enhances the ease of editing and version control for a text file. Merge conflicts are less likely to occur when small changes are made, and the changes are easier to identify.
-
Documents written with semantic line breaks are easier to navigate and edit with Vim and other text editors that use Vim keybindings.
-
Semantic line breaks are invisible to readers. The final rendered output shows no changes to the source text.
Converting existing text not written with semantic line breaks takes a long time to do it manually, and it is surprisingly difficult to do it automatically with rule-based methods.
Rule-based heuristics do not work well with the actual semantic structure of the text, often leading to incorrect semantic boundaries. Moreover, these boundaries are hierarchical and nested, and a rule-based approach cannot capture this structure. A semantic line break may occur after a dependent clause, but where to break clauses into lines is challenging to determine without syntactic and semantic reasoning capabilities. For examples:
-
A rule that breaks lines at punctuation marks will not work well with sentences that contain periods in abbreviations or mathematical expressions.
-
Syntactic or semantic structures are not always easy to determine. "I like to eat apples and oranges because they are healthy." should be broken into lines as follows:
> I like to eat apples and oranges > because they are healthy.rather than:
> I like to eat apples > and oranges because they are healthy.
For this reason, I have created SemBr, which uses finetuned Transformer models to predict line breaks at semantic boundaries.
SemBr uses a Transformer model to predict line breaks at semantic boundaries.
A small dataset of text with semantic line breaks was created from my existing LaTeX documents. The dataset was split into training (46,295 lines, 170,681 words and 1,492,952 characters) and test (2,187 lines, 7,564 words and 72,231 characters) datasets.
The data was prepared by extracting line breaks and indent levels from the files, and then converting the result into strings of paragraphs with line breaks removed. The data can then be tokenized using the tokenizer and converted into a dataset with tokens, where each token has a label denoting if there is line break before it, and the indent level of the token.
For LaTeX documents,
there are two types of line breaks:
one with a normal line break
that adds implicit spacing (e.g. line a⏎line b)
and one with no spacing (e.g. line a%⏎line b).
The data processor
also tries to preserve the LaTeX syntax of the text
by adding and removing comment symbols (%),
if necessary.
The pretrained masked language model is then finetuned as a token classifier on the training dataset to predict the labels of the tokens. We save the model with the best F1 score on correctly predicting the existence of a line break on the test set. The finetuning logs for the following models can be found on this WandB report:
distilbert-base-uncased[Pretrained] [Finetuned]distilbert-base-cased[Pretrained] [Finetuned]distilbert-base-uncased-finetuned-sst-2-english[Pretrained] [Finetuned]prajjwal1/bert-tiny[Pretrained] [Finetuned]prajjwal1/bert-mini[Pretrained] [Finetuned]prajjwal1/bert-small[Pretrained] [Finetuned]
Current inference speed on an M2 Macbook Pro
is about 850 words per second
on bert-small with the default options,
the memory usage is about 1.70 GB.
The link breaking accuracy is difficult to measure, and the locations of line breaks could also be subjective. On the test set, the per-token line break accuracy of the models are >95%, with ~80% F1 scores. Because of the sparse nature of line breaks, the accuracy is not a good metric to measure the performance of the model, and I used the F1 score instead to save best models.
- Features:
- Natural language support:
- Support natural languages other than English.
- Typesetting languages support:
-
Markdown. - Typst.
-
- Usability:
- Inference queue.
- Daemon with model unloading.
- Editor integration:
-
NeoVim plugin. -
VSCode extension. - MCP server.
-
-
Use the Hugging Face API for inference.
- Natural language support:
- Accuracy:
- Some lines are too short or too long:
- Long lines can be penalized greedily
by breaking lines with token counts
more than
optimize.preferred_max_tokens_per_line. - Support
optimize.preferred_(min|max)_words_per_line. - Improve the algorithm to penalize short and long lines with a more sophisticated method.
- Long lines can be penalized greedily
by breaking lines with token counts
more than
- Improve indent level prediction.
- Performance and accuracy benchmarking, and comparisons with related works.
- Some lines are too short or too long:
- Performance:
- Improve inference speed.
- Reduce memory usage.
Sentence splitting:
- https://code.google.com/archive/p/splitta/
- https://en.wikipedia.org/wiki/Sentence_boundary_disambiguation
- https://github.com/nipunsadvilkar/pySBD
- https://www.nltk.org/api/nltk.tokenize.sent_tokenize.html
Semantic line breaking: