Spaces:
Running
on
Zero
Running
on
Zero
prithivMLmods
commited on
Commit
•
c819544
1
Parent(s):
361acf1
Update app.py
Browse files
app.py
CHANGED
@@ -37,8 +37,8 @@ if not torch.cuda.is_available():
|
|
37 |
base_model = "black-forest-labs/FLUX.1-dev"
|
38 |
pipe = DiffusionPipeline.from_pretrained(base_model, torch_dtype=torch.bfloat16)
|
39 |
|
40 |
-
lora_repo = "prithivMLmods/
|
41 |
-
trigger_word = "
|
42 |
pipe.load_lora_weights(lora_repo)
|
43 |
|
44 |
pipe.to("cuda")
|
|
|
37 |
base_model = "black-forest-labs/FLUX.1-dev"
|
38 |
pipe = DiffusionPipeline.from_pretrained(base_model, torch_dtype=torch.bfloat16)
|
39 |
|
40 |
+
lora_repo = "prithivMLmods/Golden-Dust-Flux-LoRA"
|
41 |
+
trigger_word = "Golden Dust" # Leave trigger_word blank if not used.
|
42 |
pipe.load_lora_weights(lora_repo)
|
43 |
|
44 |
pipe.to("cuda")
|