TheImmortalPython 发布了 pklm-sandbox,这是一个轻量级的开源工具,专为在 Hugging Face transformers 中尝试自定义生成约束的开发者设计。
- 它提供了一个简单的沙盒,用于测试自定义令牌掩码管道和使用 LogitsProcessor 架构的零漂移语言控制。
- 可通过 pip 从 GitHub 仓库安装。
- 该处理器允许用户直接在标准的 generate() 管道中屏蔽特定的令牌 ID。
该项目旨在为开发者提供一个干净的起点,以便在本地实现自定义令牌控制。