prithivMLmods
commited on
Commit
•
b332ddc
1
Parent(s):
ab04f4e
Update README.md
Browse files
README.md
CHANGED
@@ -133,6 +133,21 @@ You should use `realism` to trigger the image generation.
|
|
133 |
|
134 |
You should use `face realism` to trigger the image generation.
|
135 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
136 |
|
137 |
## Download model
|
138 |
|
|
|
133 |
|
134 |
You should use `face realism` to trigger the image generation.
|
135 |
|
136 |
+
## Setting Up
|
137 |
+
```
|
138 |
+
import torch
|
139 |
+
from pipelines import DiffusionPipeline
|
140 |
+
|
141 |
+
base_model = "black-forest-labs/FLUX.1-dev"
|
142 |
+
pipe = DiffusionPipeline.from_pretrained(base_model, torch_dtype=torch.bfloat16)
|
143 |
+
|
144 |
+
lora_repo = "prithivMLmods/Canopus-LoRA-Flux-FaceRealism"
|
145 |
+
trigger_word = "Realism" # Leave trigger_word blank if not used.
|
146 |
+
pipe.load_lora_weights(lora_repo)
|
147 |
+
|
148 |
+
device = torch.device("cuda")
|
149 |
+
pipe.to(device)
|
150 |
+
```
|
151 |
|
152 |
## Download model
|
153 |
|