Photo-Restore-i2i is an adapter for black-forest-lab's FLUX.1-Kontext-dev, designed to restore old photos into mid-colorized, detailed images. The model was trained on 50 image pairs (25 start images, 25 end images). Synthetic result nodes were generated using NanoBanana from Google and SeedDream 4 (dataset for result sets), and labeled with DeepCaption-VLA-7B. The adapter is triggered with the following prompt:

[!note] [photo content], restore and enhance the image by repairing any damage, scratches, or fading. Colorize the photo naturally while preserving authentic textures and details, maintaining a realistic and historically accurate look.

Sample Inference

ex1	ex2
Left Screenshot	Right Screenshot

Parameter Settings

Setting	Value
Module Type	Adapter
Base Model	FLUX.1 Kontext Dev - fp8
Trigger Words	[photo content], restore and enhance the image by repairing any damage, scratches, or fading. Colorize the photo naturally while preserving authentic textures and details, maintaining a realistic and historically accurate look.
Image Processing Repeats	50
Epochs	28
Save Every N Epochs	1

Labeling: DeepCaption-VLA-7B(natural language & English)

Total Images Used for Training : 50 Image Pairs (25 Start, 25 End)

Synthetic Result Node generated by NanoBanana from Google (Image Result Sets Dataset)

Training Parameters

Setting	Value
Seed	-
Clip Skip	-
Text Encoder LR	0.00001
UNet LR	0.00005
LR Scheduler	constant
Optimizer	AdamW8bit
Network Dimension	64
Network Alpha	32
Gradient Accumulation Steps	-

Label Parameters

Setting	Value
Shuffle Caption	-
Keep N Tokens	-

Advanced Parameters

Setting	Value
Noise Offset	0.03
Multires Noise Discount	0.1
Multires Noise Iterations	10
Conv Dimension	-
Conv Alpha	-
Batch Size	-
Steps	4100
Sampler	euler

Trigger words

You should use [photo content] to trigger the image generation.

You should use restore and enhance the image by repairing any damage to trigger the image generation.

You should use scratches to trigger the image generation.

You should use or fading. Colorize the photo naturally while preserving authentic textures and details to trigger the image generation.

You should use maintaining a realistic and historically accurate look. to trigger the image generation.

Download model

Download them in the Files & versions tab.

prithivMLmods/Photo-Restore-i2i

Photo-Restore-i2i [Image-to-Image]