DeTikZify

环境要求

• Python 环境（需能用 pip 安装包）
• 完整安装 TeX Live 2023（用于编译 TikZ）
• 安装 ghostscript
• 安装 poppler
• 若需运行 examples，需 git 并克隆仓库

安装与启动步骤

1. 1安装系统依赖

   通过系统包管理器安装完整版 TeX Live 2023、ghostscript 与 poppler，这是编译渲染 TikZ 的必需外部依赖，缺一不可。

2. 2pip 安装 DeTikZify

   直接用 pip 从 GitHub 安装。若只用 v2/v2.5 模型可去掉 [legacy]，仅在需要 v1 模型时保留该 extra。

   pip install 'detikzify[legacy] @ git+https://github.com/potamides/DeTikZify'

   复制

3. 3克隆并可编辑安装

   想跑仓库自带 examples 时，改为此方式克隆并以可编辑模式安装 examples 依赖，便于修改代码。

   git clone https://github.com/potamides/DeTikZify
   pip install -e DeTikZify[examples]

   复制

4. 4启动 Web UI

   命令行直接启动交互式网页界面，加 --light 参数使用轻量模式，更多选项可加 --help 查看。

   python -m detikzify.webui --light

   复制

5. 5编程接口推理

   用 load 载入模型并用 DetikzifyPipeline 推理，simulate 会跑 MCTS 并逐个产出带评分的 TikZ 程序。

   python -c "from detikzify.model import load
   from detikzify.infer import DetikzifyPipeline
   pipeline = DetikzifyPipeline(*load(model_name_or_path='nllg/detikzify-v2.5-8b', device_map='auto', torch_dtype='bfloat16'))
   for score, fig in pipeline.simulate(image='https://w.wiki/A7Cc', timeout=600):
       print(score)"

   复制

关键配置

如何确认成功

执行 python -m detikzify.webui --light 后 Web UI 能正常打开，或调用 pipeline.simulate 能返回带评分的 TikZ 代码。

常见问题

注意事项

• Hugging Face Space 重启可能需要长达 30 分钟。
• 遇到长队列可复制该 Space 并使用付费私有 GPU 运行。
• TeX Live 需为完整的 2023 版本，精简安装可能无法编译。