prithivMLmods commited on
Commit
afeec04
·
verified ·
1 Parent(s): b4932d3

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +39 -0
README.md CHANGED
@@ -50,7 +50,46 @@ license: creativeml-openrail-m
50
 
51
  <Gallery />
52
 
 
53
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
54
  ## Trigger words
55
 
56
  You should use `toonic 2.5D` to trigger the image generation.
 
50
 
51
  <Gallery />
52
 
53
+ **The model is still in the training phase. This is not the final version and may contain artifacts and perform poorly in some cases.**
54
 
55
+ ## Model description
56
+
57
+ **prithivMLmods/Flux-Toonic-2.5D-LoRA**
58
+
59
+ Image Processing Parameters
60
+
61
+ | Parameter | Value | Parameter | Value |
62
+ |---------------------------|--------|---------------------------|--------|
63
+ | LR Scheduler | constant | Noise Offset | 0.03 |
64
+ | Optimizer | AdamW | Multires Noise Discount | 0.1 |
65
+ | Network Dim | 64 | Multires Noise Iterations | 10 |
66
+ | Network Alpha | 32 | Repeat & Steps | 23 & 2900 |
67
+ | Epoch | 15 | Save Every N Epochs | 1 |
68
+
69
+ Labeling: florence2-en(natural language & English)
70
+
71
+ Total Images Used for Training : 15
72
+
73
+ ## Best Dimensions
74
+
75
+ - 768 x 1024 (Best)
76
+ - 1024 x 1024 (Default)
77
+
78
+ ## Setting Up
79
+ ```python
80
+ import torch
81
+ from pipelines import DiffusionPipeline
82
+
83
+ base_model = "black-forest-labs/FLUX.1-dev"
84
+ pipe = DiffusionPipeline.from_pretrained(base_model, torch_dtype=torch.bfloat16)
85
+
86
+ lora_repo = "prithivMLmods/Flux-Toonic-2.5D-LoRA"
87
+ trigger_word = "toonic 2.5D"
88
+ pipe.load_lora_weights(lora_repo)
89
+
90
+ device = torch.device("cuda")
91
+ pipe.to(device)
92
+ ```
93
  ## Trigger words
94
 
95
  You should use `toonic 2.5D` to trigger the image generation.