介绍
# NVIDIA Image Generation
使用 NVIDIA 的 FLUX 模型生成和编辑图像。
## 模型
| 模型 | 用途 | 速度 | 质量 | |-------|----------|-------|---------| | `dev` | 高质量文本到图像 | 正常 | 最佳 | | `schnell` | 快速文本到图像 | 快 | 良好 | | `kontext` | 图像编辑 | 正常 | 最佳 |
## 快速开始
```bash # Generate an image python scripts/generate.py "A cute cat in space"
# Edit an existing image python scripts/generate.py "Add sunglasses" -i photo.jpg -o edited.png ```
## 参数
### 文本到图像 (dev/schnell)
| 参数 | 简写 | 默认值 | 描述 | |-----------|-------|---------|-------------| | `prompt` | | (必需) | 文本描述 | | `-o, --output` | | output.png | 输出文件路径 | | `--width` | | 1024 | 输出宽度（像素） | | `--height` | | 1024 | 输出高度（像素） | | `--aspect-ratio` | `-ar` | 1:1 | 纵横比预设 | | `--steps` | `-s` | 30 | 扩散步数 | | `--seed` | | 0 | 随机种子（0=随机） | | `--model` | `-m` | auto | 模型选择 |
### 图像编辑 (kontext)
| 参数 | 简写 | 默认值 | 描述 | |-----------|-------|---------|-------------| | `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 |
## 示例
### 基础生成 ```bash python scripts/generate.py "A mountain landscape at sunset" ```
### 宽屏模式 (16:9) ```bash python scripts/generate.py "A panoramic beach view" -ar 16:9 ```
### 竖屏模式 (9:16) ```bash python scripts/generate.py "A professional headshot" -ar 9:16 ```
### 自定义尺寸 ```bash python scripts/generate.py "A banner image" --width 1344 --height 768 ```
### 快速生成 ```bash python scripts/generate.py "Quick sketch of a robot" -m schnell ```
### 编辑图像 ```bash python scripts/generate.py "Make the background a sunset" -i input.jpg -o output.png ```
### 可复现结果 ```bash python scripts/generate.py "A robot" --seed 12345 ```
## 输出
脚本输出 `MEDIA:/path/to/image.png`，可直接发送到聊天中。
## API 密钥
API 密钥已嵌入脚本中。如需使用不同的密钥，请设置 `NVIDIA_API_KEY` 环境变量。