prithivMLmods commited on
Commit
275f746
1 Parent(s): 54dcc6b

Update README.md

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