prithivMLmods commited on
Commit
b79aa0d
·
verified ·
1 Parent(s): ce09cf9

Update README.md

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