Nvidia Image Gen

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.