prithivMLmods commited on
Commit
6284b2d
1 Parent(s): 56a450f

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +38 -1
README.md CHANGED
@@ -43,12 +43,49 @@ license: apache-2.0
43
 
44
  <Gallery />
45
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
46
 
47
  ## Trigger words
48
 
49
  You should use `Mockup` to trigger the image generation.
50
 
51
-
52
  ## Download model
53
 
54
  Weights for this model are available in Safetensors format.
 
43
 
44
  <Gallery />
45
 
46
+ **The model is still in the training phase. This is not the final version and may contain artifacts and perform poorly in some cases.**
47
+
48
+ ## Model description
49
+
50
+ **prithivMLmods/Mockup-Texture-Flux-LoRA**
51
+
52
+ Image Processing Parameters
53
+
54
+ | Parameter | Value | Parameter | Value |
55
+ |---------------------------|--------|---------------------------|--------|
56
+ | LR Scheduler | constant | Noise Offset | 0.03 |
57
+ | Optimizer | AdamW | Multires Noise Discount | 0.1 |
58
+ | Network Dim | 64 | Multires Noise Iterations | 10 |
59
+ | Network Alpha | 32 | Repeat & Steps | 23 & 2.2K |
60
+ | Epoch | 13 | Save Every N Epochs | 1 |
61
+
62
+ Labeling: florence2-en(natural language & English)
63
+
64
+ Total Images Used for Training : 22 [ Hi-RES ]
65
+
66
+ ## Setting Up
67
+ ```
68
+ import torch
69
+ from pipelines import DiffusionPipeline
70
+
71
+ base_model = "black-forest-labs/FLUX.1-dev"
72
+ pipe = DiffusionPipeline.from_pretrained(base_model, torch_dtype=torch.bfloat16)
73
+
74
+ lora_repo = "prithivMLmods/Mockup-Texture-Flux-LoRA"
75
+ trigger_word = "Mockup"
76
+ pipe.load_lora_weights(lora_repo)
77
+
78
+ device = torch.device("cuda")
79
+ pipe.to(device)
80
+ ```
81
+ ## Data source
82
+
83
+ - https://playground.com/
84
 
85
  ## Trigger words
86
 
87
  You should use `Mockup` to trigger the image generation.
88
 
 
89
  ## Download model
90
 
91
  Weights for this model are available in Safetensors format.