prithivMLmods commited on
Commit
3db4bd6
·
verified ·
1 Parent(s): 82989f2

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +41 -0
README.md CHANGED
@@ -22,6 +22,47 @@ license: creativeml-openrail-m
22
 
23
  <Gallery />
24
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
25
  ## Trigger words
26
 
27
  You should use `2color illustration` to trigger the image generation.
 
22
 
23
  <Gallery />
24
 
25
+ # Model description for 2Color-Illustration
26
+ Image Processing Parameters
27
+
28
+ | Parameter | Value | Parameter | Value |
29
+ |---------------------------|--------|---------------------------|--------|
30
+ | LR Scheduler | constant | Noise Offset | 0.03 |
31
+ | Optimizer | AdamW | Multires Noise Discount | 0.1 |
32
+ | Network Dim | 64 | Multires Noise Iterations | 10 |
33
+ | Network Alpha | 32 | Repeat & Steps | 24 & 2770 |
34
+ | Epoch | 20 | Save Every N Epochs | 1 |
35
+
36
+ Labeling: florence2-en(natural language & English)
37
+
38
+ Total Images Used for Training : 22
39
+
40
+ ## Best Dimensions & Inference
41
+
42
+ | **Dimensions** | **Aspect Ratio** | **Recommendation** |
43
+ |-----------------|------------------|---------------------------|
44
+ | 1280 x 832 | 3:2 | Best |
45
+ | 1024 x 1024 | 1:1 | Default |
46
+
47
+ ### Inference Range
48
+
49
+ - **Recommended Inference Steps:** 30–35
50
+
51
+ ## Setting Up
52
+ ```python
53
+ import torch
54
+ from pipelines import DiffusionPipeline
55
+
56
+ base_model = "black-forest-labs/FLUX.1-dev"
57
+ pipe = DiffusionPipeline.from_pretrained(base_model, torch_dtype=torch.bfloat16)
58
+
59
+ lora_repo = "strangerzonehf/2Color-Illustration"
60
+ trigger_word = "2color illustration"
61
+ pipe.load_lora_weights(lora_repo)
62
+
63
+ device = torch.device("cuda")
64
+ pipe.to(device)
65
+ ```
66
  ## Trigger words
67
 
68
  You should use `2color illustration` to trigger the image generation.