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.