metadata

library_name: transformers
tags: []

Model Card for TikZero+ (10b)

TikZero+ (10b) is a multimodal language model that automatically synthesizes scientific figures as editable, semantics-preserving TikZ graphics programs conditioned on text captions. It is based on DeTikZify_v2 (8b) and LLaMA_3.2 (1b). Check out the DeTikZify project for more information and tips on how to best run the model.

Usage

from detikzify.model import load
from detikzify.infer import DetikzifyPipeline

caption = "A multi-layer perceptron with two hidden layers."
pipeline = DetikzifyPipeline(*load(
    model_name_or_path="nllg/tikzero-plus-10b",
    device_map="auto",
    torch_dtype="bfloat16",
))

# generate a single TikZ program
fig = pipeline.sample(text=caption)

# if it compiles, rasterize it and show it
if fig.is_rasterizable:
    fig.rasterize().show()