prithivMLmods commited on
Commit
879a962
1 Parent(s): 5fe98c6

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +42 -0
README.md CHANGED
@@ -47,6 +47,48 @@ license: apache-2.0
47
 
48
  <Gallery />
49
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
50
  ## Trigger words
51
 
52
  You should use `Sketch Flat` to trigger the image generation.
 
47
 
48
  <Gallery />
49
 
50
+ # Model description for Flux-Sketch-Flat-LoRA
51
+
52
+ Image Processing Parameters
53
+
54
+ | Parameter | Value | Parameter | Value |
55
+ |---------------------------|--------|---------------------------|--------|
56
+ | LR Scheduler | constant | Noise Offset | 0.03 |
57
+ | Optimizer | AdamW | Multires Noise Discount | 0.1 |
58
+ | Network Dim | 64 | Multires Noise Iterations | 10 |
59
+ | Network Alpha | 32 | Repeat & Steps | 28 & 3990 |
60
+ | Epoch | 20 | Save Every N Epochs | 1 |
61
+
62
+ Labeling: florence2-en(natural language & English)
63
+
64
+ Total Images Used for Training : 42
65
+
66
+ ## Best Dimensions & Inference
67
+
68
+ | **Dimensions** | **Aspect Ratio** | **Recommendation** |
69
+ |-----------------|------------------|---------------------------|
70
+ | 1280 x 832 | 3:2 | Best |
71
+ | 1024 x 1024 | 1:1 | Default |
72
+
73
+ ### Inference Range
74
+
75
+ - **Recommended Inference Steps:** 30–35
76
+
77
+ ## Setting Up
78
+ ```python
79
+ import torch
80
+ from pipelines import DiffusionPipeline
81
+
82
+ base_model = "black-forest-labs/FLUX.1-dev"
83
+ pipe = DiffusionPipeline.from_pretrained(base_model, torch_dtype=torch.bfloat16)
84
+
85
+ lora_repo = "strangerzonehf/Flux-Sketch-Flat-LoRA"
86
+ trigger_word = "Sketch Flat"
87
+ pipe.load_lora_weights(lora_repo)
88
+
89
+ device = torch.device("cuda")
90
+ pipe.to(device)
91
+ ```
92
  ## Trigger words
93
 
94
  You should use `Sketch Flat` to trigger the image generation.