prithivMLmods commited on
Commit
226ec03
·
verified ·
1 Parent(s): c5ef2f9

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +42 -1
README.md CHANGED
@@ -67,11 +67,52 @@ license: apache-2.0
67
 
68
  <Gallery />
69
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
70
  ## Trigger words
71
 
72
  You should use `Whaaattt Art` to trigger the image generation.
73
 
74
-
75
  ## Download model
76
 
77
  Weights for this model are available in Safetensors format.
 
67
 
68
  <Gallery />
69
 
70
+ # Model description for Whaaaattttt
71
+
72
+ Image Processing Parameters
73
+
74
+ | Parameter | Value | Parameter | Value |
75
+ |---------------------------|--------|---------------------------|--------|
76
+ | LR Scheduler | constant | Noise Offset | 0.03 |
77
+ | Optimizer | AdamW | Multires Noise Discount | 0.1 |
78
+ | Network Dim | 64 | Multires Noise Iterations | 10 |
79
+ | Network Alpha | 32 | Repeat & Steps | 20 & 4900 |
80
+ | Epoch | 33 | Save Every N Epochs | 1 |
81
+
82
+ Labeling: florence2-en(natural language & English)
83
+
84
+ Total Images Used for Training : 104
85
+
86
+ ## Best Dimensions & Inference
87
+
88
+ | **Dimensions** | **Aspect Ratio** | **Recommendation** |
89
+ |-----------------|------------------|---------------------------|
90
+ | 1280 x 832 | 3:2 | Best |
91
+ | 1024 x 1024 | 1:1 | Default |
92
+
93
+ ### Inference Range
94
+
95
+ - **Recommended Inference Steps:** 30–35
96
+
97
+ ## Setting Up
98
+ ```python
99
+ import torch
100
+ from pipelines import DiffusionPipeline
101
+
102
+ base_model = "black-forest-labs/FLUX.1-dev"
103
+ pipe = DiffusionPipeline.from_pretrained(base_model, torch_dtype=torch.bfloat16)
104
+
105
+ lora_repo = "strangerzonehf/Whaaaattttt"
106
+ trigger_word = "Whaaattt Art"
107
+ pipe.load_lora_weights(lora_repo)
108
+
109
+ device = torch.device("cuda")
110
+ pipe.to(device)
111
+ ```
112
  ## Trigger words
113
 
114
  You should use `Whaaattt Art` to trigger the image generation.
115
 
 
116
  ## Download model
117
 
118
  Weights for this model are available in Safetensors format.