You're looking at a specific version of this model. Jump to the model overview.
simpletuner /qwen-image:33f2f901
Input schema
The fields you can use to run this model with an API. If you don’t give a value for a field its default value will be used.
| Field | Type | Default value | Description |
|---|---|---|---|
| train_data |
string
|
Zip or tar of training images with matching .txt captions.
|
|
| hf_dataset |
string
|
Hugging Face dataset repo ID to use when no archive is provided (e.g., user/dataset).
|
|
| model |
None
|
Qwen-Image-2.1
|
Base model to train. Qwen Image 2.1 uses the 80 GB assisted REPA recipe.
|
| resolution |
None
|
512+1024
|
Training and validation scales; multi-scale sampling preserves aspect ratios.
|
| use_synthetic_regularisation |
boolean
|
True
|
Qwen-Image-2.1: mix synthetic images into 50% of updates to reduce concept bleed into unrelated prompts. Disable to give all updates to your images; roughly half as many total steps provide the same number of concept updates, but bleed increases. Has no effect on original Qwen-Image.
|
| hf_caption_column |
string
|
caption
|
Caption column when using hf_dataset.
|
| validation_prompt |
string
|
a detailed photograph, natural lighting
|
Validation prompt; include the concept or trigger you are teaching.
|
| enable_crop |
boolean
|
False
|
Enable cropping when preparing samples.
|
| crop_style |
None
|
center
|
Crop location style when cropping is enabled.
|
| crop_aspect |
None
|
preserve
|
Aspect handling for crops: square (1:1) or preserve source aspect.
|
| train_batch_size |
integer
|
1
Min: 1 Max: 4 |
Training batch size (1-4).
|
| checkpoints_total_limit |
integer
|
5
Min: 1 Max: 5 |
Maximum checkpoints to retain (max 5).
|
| checkpoint_step_interval |
integer
|
500
Min: 1 |
Save checkpoints and generate validation images every N optimizer updates.
|
| max_train_steps |
integer
|
4000
Min: 1 |
Total optimizer updates; the multi-scale 80 GB recipe starts at 4000.
|
| s3_bucket |
string
|
|
S3-compatible bucket for publishing checkpoints.
|
| s3_region |
string
|
S3 region (optional).
|
|
| s3_endpoint_url |
string
|
Custom S3 endpoint URL (for non-AWS providers).
|
|
| s3_base_path |
string
|
Prefix inside the bucket (defaults to cog/qwen-image/{job_id}).
|
|
| s3_public_base_url |
string
|
Public base URL to build shareable links (optional).
|
|
| s3_access_key |
string
|
S3 access key (leave blank to use IAM/instance roles).
|
|
| s3_secret_key |
string
|
S3 secret key (leave blank to use IAM/instance roles).
|
|
| hub_model_id |
string
|
Hugging Face model repo for the final LoRA and validation images (e.g., user/model).
|
|
| hf_token |
string
|
Hugging Face token for downloads and write access when publishing to hub_model_id.
|
|
| return_logs |
boolean
|
True
|
Print the tail of debug.log.
|
Output schema
The shape of the response you’ll get when you run this model with an API.
Schema
{'format': 'uri', 'title': 'Output', 'type': 'string'}