dvcrn NVIDIA Image Generation
Generate and edit images using NVIDIA's FLUX models.
Models
| Model | Use Case | Speed | Quality |
|---|---|---|---|
dev |
High-quality text-to-image | Normal | Best |
schnell |
Fast text-to-image | Fast | Good |
kontext |
Image editing | Normal | Best |
Quick Start
# 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
Parameters
Text-to-Image (dev/schnell)
| Parameter | Short | Default | Description |
|---|---|---|---|
prompt |
(required) | Text description | |
-o, --output |
output.png | Output file path | |
--width |
1024 | Output width in pixels | |
--height |
1024 | Output height in pixels | |
--aspect-ratio |
-ar |
1:1 | Aspect ratio preset |
--steps |
-s |
30 | Diffusion steps |
--seed |
0 | Random seed (0=random) | |
--model |
-m |
auto | Model selection |
Image Editing (kontext)
| Parameter | Short | Default | Description |
|---|---|---|---|
prompt |
(required) | Edit instruction | |
-i, --input |
(required) | Input image path | |
-o, --output |
output.png | Output file path | |
--steps |
-s |
30 | Diffusion steps |
--cfg |
3.5 | Guidance scale | |
--seed |
0 | Random seed |
Supported Aspect Ratios
| Ratio | Resolution |
|---|---|
| 1:1 | 1024×1024 |
| 16:9 | 1344×768 |
| 9:16 | 768×1344 |
| 4:3 | 1216×832 |
| 3:4 | 832×1216 |
Examples
Basic Generation
python scripts/generate.py "A mountain landscape at sunset"
Wide Format (16:9)
python scripts/generate.py "A panoramic beach view" -ar 16:9
Portrait Mode (9:16)
python scripts/generate.py "A professional headshot" -ar 9:16
Custom Size
python scripts/generate.py "A banner image" --width 1344 --height 768
Fast Generation
python scripts/generate.py "Quick sketch of a robot" -m schnell
Edit an Image
python scripts/generate.py "Make the background a sunset" -i input.jpg -o output.png
Reproducible Results
python scripts/generate.py "A robot" --seed 12345
Output
The script outputs MEDIA:/path/to/image.png which can be sent directly to chat.
API Key
The API key is embedded in the script. To use a different key, set the NVIDIA_API_KEY environment variable.