名称： nvidia-image-gen
版本： 1.0.0
描述： 使用 NVIDIA FLUX 模型生成和编辑图像。当用户要求生成图像、创建图片、编辑照片或用 AI 修改现有图像时使用。支持基于文本提示的文生图和图像编辑。
使用 NVIDIA 的 FLUX 模型生成和编辑图像。
| 模型 | 用途 | 速度 | 质量 |
|---|---|---|---|
dev |
高质量文生图 | 标准 | 最佳 |
schnell |
快速文生图 | 快速 | 良好 |
kontext |
图像编辑 | 标准 | 最佳 |
# 生成图像
python scripts/generate.py "一只太空中的可爱猫咪"
# 编辑现有图像
python scripts/generate.py "添加太阳镜" -i photo.jpg -o edited.png
| 参数 | 简写 | 默认值 | 描述 |
|---|---|---|---|
prompt |
(必填) | 文本描述 | |
-o, --output |
output.png | 输出文件路径 | |
--width |
1024 | 输出宽度（像素） | |
--height |
1024 | 输出高度（像素） | |
--aspect-ratio |
-ar |
1:1 | 预设宽高比 |
--steps |
-s |
30 | 扩散步数 |
--seed |
0 | 随机种子 (0=随机) | |
--model |
-m |
auto | 模型选择 |
| 参数 | 简写 | 默认值 | 描述 |
|---|---|---|---|
prompt |
(必填) | 编辑指令 | |
-i, --input |
(必填) | 输入图像路径 | |
-o, --output |
output.png | 输出文件路径 | |
--steps |
-s |
30 | 扩散步数 |
--cfg |
3.5 | 引导尺度 | |
--seed |
0 | 随机种子 |
| 比例 | 分辨率 |
|---|---|
| 1:1 | 1024×1024 |
| 16:9 | 1344×768 |
| 9:16 | 768×1344 |
| 4:3 | 1216×832 |
| 3:4 | 832×1216 |
python scripts/generate.py "日落时的山景"
python scripts/generate.py "全景海滩风光" -ar 16:9
python scripts/generate.py "专业头像照" -ar 9:16
python scripts/generate.py "横幅图像" --width 1344 --height 768
python scripts/generate.py "机器人的快速草图" -m schnell
python scripts/generate.py "将背景改为日落" -i input.jpg -o output.png
python scripts/generate.py "一个机器人" --seed 12345
脚本输出 MEDIA:/path/to/image.png，可直接发送至聊天。
API 密钥已嵌入脚本中。如需使用其他密钥，请设置 NVIDIA_API_KEY 环境变量。