prithivMLmods commited on
Commit
cd9c3bf
·
verified ·
1 Parent(s): ef213f5

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +44 -0
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.