Midjourney Image Generation with Wavespeed API
A Cog-based model that generates images using the Midjourney API via Wavespeed AI.
Overview
This project wraps the Wavespeed AI Midjourney text-to-image API into a deployable Cog model. It provides a clean interface for generating AI images with customizable parameters like aspect ratio, quality, style, and more.
Features
- 🎨 Midjourney Integration - Uses Wavespeed AI’s Midjourney API
- ⚙️ Flexible Configuration - Customize aspect ratio, quality, chaos, stylize, and other parameters
- 🎭 Multiple Model Versions - Support for Midjourney v6, v6.1, and v7
- 🌟 Niji Support - Built-in support for Niji anime model variants
- 🔄 Async Job Processing - Automatically polls for task completion
Installation
Input Parameters
| Parameter | Type | Description | Default |
|---|---|---|---|
prompt |
str | Text description for image generation | ”“ |
aspect_ratio |
str | Image aspect ratio (9:16, 1:1, 16:9, etc.) | “9:16” |
quality |
float | Image quality (0.25, 0.5, 1, 2) | 1 |
version |
str | Midjourney version (“6”, “6.1”, “7”) | “7” |
niji |
str | Niji model version (“0”, “5”, “6”, “close”) | “close” |
seed |
int | Random seed for generation (-1 for random) | -1 |
chaos |
float | Chaos level (0-100) | 0 |
stylize |
float | Stylization level (0-1000) | 0 |
weird |
float | Weirdness factor (0-3000) | 0 |
enable_base64_output |
bool | Enable base64 output | False |
wavespeed_api_token |
str | Wavespeed API token (secret) | ”“ |
Configuration
Set your Wavespeed API key either:
- As an input parameter when calling the model
- As the WAVESPEED_API_KEY environment variable
Output
Returns a PNG image file generated by Midjourney based on your input parameters.
Model created