pjit-21k

pjit-21k

Linum 的实验性 2B 文生图模型，训练算力约 2.1 万美元，最适合用 50 到 80 个词的详细提示词，不能在图里写字。适合研究人员和好奇的创作者。

由 Linum-AI 在 Hugging Face 上制作。这些说明是 Mindket 根据它的模型页面和模型卡写的，模型自己的文件和许可证在 Hugging Face 上。

简介

它是什么：pjit-21k，也叫 P-JiT 21K，是 Linum 的文生图模型，训练算力花了约 2.1 万美元。它是一个 2B 参数的扩散 transformer，搭配一个冻结的 Qwen3.5-4B 文本编码器，后者需要单独下载。Linum 称它是研究检查点，不是生产模型，并提醒会有粗糙之处。它见过 6.6 亿个训练样本，之后还有一步后训练。

适合谁：研究图像模型高效训练的研究人员，以及想用详细描述来试一个小型实验模型的创作者。

Industry | Who uses it | How they use it | Example
研究 | 图像模型研究人员 | 研究一个低成本训练的结果 | 阅读代码仓库并复现一个样例
创意工作 | 插画师 | 根据详细描述生成图片 | 提示一个小熊和兔子的儿童书场景
软件 | 开发者 | 测试一个小型文生图模型 | 带着提示词运行生成脚本

如果你需要在图片里写字，可以跳过它。它不能写文字、海报、标志或招牌，因为 Linum 把渲染文字从训练数据里过滤掉了。

用在哪里：

• 带有项目代码仓库的 Python 环境，里面有生成脚本
• OpenRouter，只有使用可选的提示词扩写器时才需要

Use case | Who | What they get back
用详细提示词生成图片 | 艺术家 | 根据 50 到 80 个词的描述生成的图片
用简短提示词生成图片 | 普通用户 | 扩写器改写提示词后得到的图片
训练研究 | 研究人员 | 一个小型模型检查点及可供查看的代码

如何设置：

• Python，以及 GitHub 上同名的代码仓库
• 能运行 2.18B 参数模型的硬件，fp32 下 8.7 GB，外加单独的 4B 文本编码器
• OPENROUTER_API_KEY，只有想用提示词扩写器时才需要

1. 获取项目的代码仓库并进入其中。
2. 以可编辑模式安装。
3. 带着提示词和种子运行生成脚本。

pip install -e .

pip install -U huggingface_hub
hf download Linum-AI/pjit-21k

如何使用：写一段 50 到 80 个词的详细提示词，或者让扩写器把简短的提示词展开。用 no upsample 参数可以关掉扩写器。

export OPENROUTER_API_KEY=...
python generate.py --prompt "oil painting of weathered fisherman smoking a pipe" --aspect_ratio square --seeds 0

Tool | What it does
generate.py | 根据提示词生成图片
--no_upsample | 关闭提示词扩写器
--aspect_ratio | 选择画面形状，比如 tall 或 square
--seeds | 设置随机种子
OPENROUTER_API_KEY | 扩写器需要的密钥

示例：设置好密钥后运行上面的命令。扩写器会把简短的提示词展开成详细的描述，模型返回一幅油画风格的、抽着烟斗的饱经风霜的渔夫图片。

新手与专家：如果你是新手，写一段描述场景的详细文字，并带上 no upsample 参数运行，这样就不会用到外部服务。

如果你熟悉这个领域，基础设计是 P-JiT。Linum 训练得更久，然后用一种 FDFO 做了一轮后训练。采样默认值是 50 步 Euler 和缩放为 15 的自适应投影引导。它在五种约一百万像素的尺寸上训练过。

输入一段详细的描述
冻结的 Qwen3.5-4B 对它编码
2B 扩散 transformer 去噪
输出一张图片

Title: 按图片尺寸统计的训练样本
512px: 610 million
1024px: 50 million

注意事项：

• 它不擅长全身照、同时出现很多物体、多人互动，以及世界知识，比如吉他有几根弦。
• 检查点文件里放的是 2.18B 参数的 transformer，不含仅用于训练的头。
• License: apache-2.0。
• Cost to run: 可选的扩写器需要 OpenRouter 密钥，它通过 OpenRouter 调用 Claude。模型本身没有提到费用。
• Data it reaches: 使用扩写器时，你的提示词会发给 OpenRouter。文本编码器需要单独下载。
• Not for: 按模型卡所说，不适合文字、海报、标志或生产使用。

Hugging Face 上的数据（2026年11月13日）

• 任务：text to image
• 框架：pjit-21k
• 参数量：2.2B
• Weights: about 8.7 GB in F32; about 1.1 GB at 4-bit (estimate)
• Formats: Safetensors
• 最近 30 天下载：623
• 点赞：73
• 许可证：apache-2.0
• Quantized versions: 1
• Fine-tunes: 4
• 标签：pjit-21k, safetensors, text-to-image, diffusion, pixel-space, research

链接

• 模型：https://huggingface.co/Linum-AI/pjit-21k
• 文件：https://huggingface.co/Linum-AI/pjit-21k/tree/main

获取模型

用 Hugging Face 的命令行下载它的文件：

pip install -U huggingface_hub
hf download Linum-AI/pjit-21k

然后按它的模型卡使用它。这个工具包里没有会自己运行的东西。