prithivMLmods
commited on
Update README.md
Browse files
README.md
CHANGED
@@ -122,6 +122,50 @@ license: creativeml-openrail-m
|
|
122 |
|
123 |
<Gallery />
|
124 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
125 |
## Trigger words
|
126 |
|
127 |
You should use `flat lay` to trigger the image generation.
|
|
|
122 |
|
123 |
<Gallery />
|
124 |
|
125 |
+
# **Model description for Flux-FlatLay-LoRA**
|
126 |
+
|
127 |
+
Image Processing Parameters
|
128 |
+
|
129 |
+
| Parameter | Value | Parameter | Value |
|
130 |
+
|---------------------------|--------|---------------------------|--------|
|
131 |
+
| LR Scheduler | constant | Noise Offset | 0.03 |
|
132 |
+
| Optimizer | AdamW | Multires Noise Discount | 0.1 |
|
133 |
+
| Network Dim | 64 | Multires Noise Iterations | 10 |
|
134 |
+
| Network Alpha | 32 | Repeat & Steps | 28 & 3900 |
|
135 |
+
| Epoch | 23 | Save Every N Epochs | 1 |
|
136 |
+
|
137 |
+
Labeling: florence2-en(natural language & English)
|
138 |
+
|
139 |
+
Total Images Used for Training : 60 [ 14 bit raw ]
|
140 |
+
|
141 |
+
## Best Dimensions & Inference
|
142 |
+
|
143 |
+
| **Dimensions** | **Aspect Ratio** | **Recommendation** |
|
144 |
+
|-----------------|------------------|---------------------------|
|
145 |
+
| 1280 x 832 | 3:1 | Compatible |
|
146 |
+
| 1024 x 1024 | 768*1024 | Best |
|
147 |
+
| 1024 x 1024 | 1:1 | Default |
|
148 |
+
|
149 |
+
### Inference Range
|
150 |
+
|
151 |
+
- **Recommended Inference Steps:** 30–35
|
152 |
+
|
153 |
+
## Setting Up
|
154 |
+
```python
|
155 |
+
import torch
|
156 |
+
from pipelines import DiffusionPipeline
|
157 |
+
|
158 |
+
base_model = "black-forest-labs/FLUX.1-dev"
|
159 |
+
pipe = DiffusionPipeline.from_pretrained(base_model, torch_dtype=torch.bfloat16)
|
160 |
+
|
161 |
+
lora_repo = "strangerzonehf/Flux-FlatLay-LoRA"
|
162 |
+
trigger_word = "flat lay"
|
163 |
+
pipe.load_lora_weights(lora_repo)
|
164 |
+
|
165 |
+
device = torch.device("cuda")
|
166 |
+
pipe.to(device)
|
167 |
+
```
|
168 |
+
|
169 |
## Trigger words
|
170 |
|
171 |
You should use `flat lay` to trigger the image generation.
|