prithivMLmods commited on
Commit
151b657
·
verified ·
1 Parent(s): 5bb62ec

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +39 -1
README.md CHANGED
@@ -31,11 +31,49 @@ license: creativeml-openrail-m
31
 
32
  <Gallery />
33
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
34
  ## Trigger words
35
 
36
  You should use `sketch cx` to trigger the image generation.
37
 
38
-
39
  ## Download model
40
 
41
  Weights for this model are available in Safetensors format.
 
31
 
32
  <Gallery />
33
 
34
+ # **Model description for Flux-Cx-Sketch-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 | 23 & 2780 |
44
+ | Epoch | 18 | Save Every N Epochs | 1 |
45
+
46
+ Labeling: florence2-en(natural language & English)
47
+
48
+ Total Images Used for Training : 29
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-Cx-Sketch-LoRA"
67
+ trigger_word = "sketch cx"
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 `sketch cx` to trigger the image generation.
76
 
 
77
  ## Download model
78
 
79
  Weights for this model are available in Safetensors format.