prithivMLmods commited on
Commit
a476827
1 Parent(s): 38910fa

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +35 -1
README.md CHANGED
@@ -22,12 +22,46 @@ license: apache-2.0
22
 
23
  <Gallery />
24
 
 
25
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
26
  ## Trigger words
27
 
28
  You should use `Animeo` to trigger the image generation.
29
 
30
-
31
  ## Download model
32
 
33
  Weights for this model are available in Safetensors format.
 
22
 
23
  <Gallery />
24
 
25
+ ## Model description
26
 
27
+ Image Processing Parameters
28
+
29
+ | Parameter | Value | Parameter | Value |
30
+ |---------------------------|--------|---------------------------|--------|
31
+ | LR Scheduler | constant | Noise Offset | 0.03 |
32
+ | Optimizer | AdamW | Multires Noise Discount | 0.1 |
33
+ | Network Dim | 64 | Multires Noise Iterations | 10 |
34
+ | Network Alpha | 32 | Repeat & Steps | 35 & 4700|
35
+ | Epoch | 22 | Save Every N Epochs | 1 |
36
+
37
+ Labeling: florence2-en(natural language & English)
38
+
39
+ Total Images Used for Training : 50 [ 10@5 ] [ Hi -Res ]
40
+
41
+ ## Best Dimensions
42
+
43
+ - 768 x 1024 (Best)
44
+ - 1024 x 1024 (Default)
45
+
46
+ ## Setting Up
47
+ ```python
48
+ import torch
49
+ from pipelines import DiffusionPipeline
50
+
51
+ base_model = "black-forest-labs/FLUX.1-dev"
52
+ pipe = DiffusionPipeline.from_pretrained(base_model, torch_dtype=torch.bfloat16)
53
+
54
+ lora_repo = "strangerzonehf/Flux-Animeo-v1-LoRA"
55
+ trigger_word = "Animeo"
56
+ pipe.load_lora_weights(lora_repo)
57
+
58
+ device = torch.device("cuda")
59
+ pipe.to(device)
60
+ ```
61
  ## Trigger words
62
 
63
  You should use `Animeo` to trigger the image generation.
64
 
 
65
  ## Download model
66
 
67
  Weights for this model are available in Safetensors format.