prithivMLmods commited on
Commit
313f0cb
β€’
1 Parent(s): 0c0ef50

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +44 -0
README.md CHANGED
@@ -16,6 +16,50 @@ license: creativeml-openrail-m
16
  |-----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|
17
  | World of GTA 6, Captured at eye-level on a high-angle shot of a highway at sunset. The sky is a vibrant pink, with a few wispy clouds scattered across the sky. The sun is setting, casting a warm glow over the scene. In the foreground, a bridge stretches over a body of water, dotted with trees and a few cars. The bridge is adorned with a white truck with a red sign on the side, adding a pop of color to the scene to the image, GTA 6 Theme |
18
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
19
 
20
  ## Trigger words
21
 
 
16
  |-----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|
17
  | World of GTA 6, Captured at eye-level on a high-angle shot of a highway at sunset. The sky is a vibrant pink, with a few wispy clouds scattered across the sky. The sun is setting, casting a warm glow over the scene. In the foreground, a bridge stretches over a body of water, dotted with trees and a few cars. The bridge is adorned with a white truck with a red sign on the side, adding a pop of color to the scene to the image, GTA 6 Theme |
18
 
19
+ **The model is still in the training phase. This is not the final version and may contain artifacts and perform poorly in some cases.**
20
+
21
+ ## Model description
22
+
23
+ **prithivMLmods/Castor-Character-Polygon-LoRA**
24
+
25
+ Image Processing Parameters
26
+
27
+ | Parameter | Value | Parameter | Value |
28
+ |---------------------------|--------|---------------------------|--------|
29
+ | LR Scheduler | constant | Noise Offset | 0.03 |
30
+ | Optimizer | AdamW | Multires Noise Discount | 0.1 |
31
+ | Network Dim | 64 | Multires Noise Iterations | 10 |
32
+ | Network Alpha | 32 | Repeat & Steps | 15 & 1K |
33
+ | Epoch | 10 | Save Every N Epochs | 1 |
34
+
35
+ Labeling: florence2-en(natural language & English)
36
+
37
+ Total Images Used for Training : 15+ [ Hi-RES ]
38
+
39
+ ## Setting Up
40
+ ```
41
+ import torch
42
+ from pipelines import DiffusionPipeline
43
+
44
+ base_model = "black-forest-labs/FLUX.1-dev"
45
+ pipe = DiffusionPipeline.from_pretrained(base_model, torch_dtype=torch.bfloat16)
46
+
47
+ lora_repo = "prithivMLmods/Castor-Gta6-Theme-Flux-LoRA"
48
+ trigger_word = "World of GTA 6, GTA 6 Theme" # Leave trigger_word blank if not used.
49
+ pipe.load_lora_weights(lora_repo)
50
+
51
+ device = torch.device("cuda")
52
+ pipe.to(device)
53
+ ```
54
+ ## App File Structure
55
+
56
+ /project-root/
57
+
58
+ β”œβ”€β”€ .gitattributes
59
+ β”œβ”€β”€ README.md
60
+ β”œβ”€β”€ app.py
61
+ β”œβ”€β”€ pythonproject.py
62
+
63
 
64
  ## Trigger words
65