Spaces:
Running
on
Zero
Running
on
Zero
prithivMLmods
commited on
Commit
•
a3096a3
1
Parent(s):
0183f6a
Update app.py
Browse files
app.py
CHANGED
@@ -28,8 +28,8 @@ if not torch.cuda.is_available():
|
|
28 |
base_model = "black-forest-labs/FLUX.1-dev"
|
29 |
pipe = DiffusionPipeline.from_pretrained(base_model, torch_dtype=torch.bfloat16)
|
30 |
|
31 |
-
lora_repo = "prithivMLmods/Canopus-
|
32 |
-
trigger_word = "
|
33 |
pipe.load_lora_weights(lora_repo)
|
34 |
|
35 |
pipe.to("cuda")
|
|
|
28 |
base_model = "black-forest-labs/FLUX.1-dev"
|
29 |
pipe = DiffusionPipeline.from_pretrained(base_model, torch_dtype=torch.bfloat16)
|
30 |
|
31 |
+
lora_repo = "prithivMLmods/Canopus-LoRA-Flux-FaceRealism"
|
32 |
+
trigger_word = "Realism" # Leave trigger_word blank if not used.
|
33 |
pipe.load_lora_weights(lora_repo)
|
34 |
|
35 |
pipe.to("cuda")
|