Photo-Restore-i2i / README.md
prithivMLmods's picture
Update README.md
27c4224 verified
metadata
tags:
  - text-to-image
  - lora
  - diffusers
  - template:diffusion-lora
  - photo
  - restore
  - art
widget:
  - src: images/1.jpg
    text: >-
      [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.
    prompt: >
      [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..
    output:
      url: images/2.webp
base_model: black-forest-labs/FLUX.1-Kontext-dev
instance_prompt: >-
  [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.
license: other
license_name: flux-1-dev-non-commercial-license
license_link: LICENSE.md
pipeline_tag: image-to-image

1.png

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

Prompt
[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.

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:

[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.