prithivMLmods commited on
Commit
b332ddc
1 Parent(s): ab04f4e

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +15 -0
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