HaaDeej commited on
Commit
a336387
·
verified ·
1 Parent(s): 43c700d

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +20 -0
README.md ADDED
@@ -0,0 +1,20 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: other
3
+ base_model: black-forest-labs/FLUX.2-klein-4B
4
+ tags: [lora, colorization, flux2]
5
+ ---
6
+ # Flux2-3500K — FLUX.2-klein-4B colorize LoRA (runD, 3500 steps)
7
+
8
+ Grayscale -> color img2img LoRA voor FLUX.2-klein-4B. Beste checkpoint ("runD-3500").
9
+
10
+ **Recept:** rank 32, lora_alpha 32, lr 1e-4 cosine (warmup 200), weighting_scheme logit_normal (mean -1.0), guidance 1.0 (train=eval), 3500 steps, resolutie 512, img2img (grijs = conditioning). Getraind op recipe_A modern color/gray paren + Spiegel v1-anchor.
11
+
12
+ **Gebruik:**
13
+ ```python
14
+ from diffusers import Flux2KleinPipeline
15
+ import torch
16
+ pipe = Flux2KleinPipeline.from_pretrained('black-forest-labs/FLUX.2-klein-4B', torch_dtype=torch.bfloat16).to('cuda')
17
+ pipe.load_lora_weights('HaaDeej/Flux2-3500K')
18
+ out = pipe(image=gray_square, prompt_embeds=cached_embed, num_inference_steps=4, guidance_scale=1.0).images[0]
19
+ ```
20
+ Prompt: "colorize this black and white photograph with natural, realistic colors" (~4 steps, ~3.5s/frame op RTX 4090).