Spaces:
Running
on
Zero
Running
on
Zero
prithivMLmods
commited on
Commit
•
6ae9433
1
Parent(s):
82e6520
Update app.py
Browse files
app.py
CHANGED
@@ -37,11 +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 = "strangerzonehf/Flux-
|
41 |
-
trigger_word = "
|
42 |
-
|
43 |
-
#lora_repo = "strangerzonehf/Flux-Super-Realism-LoRA"
|
44 |
-
#trigger_word = "Super Realism" # Leave trigger_word blank if not used.
|
45 |
|
46 |
pipe.load_lora_weights(lora_repo)
|
47 |
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 = "strangerzonehf/Flux-Super-Realism-LoRA"
|
41 |
+
trigger_word = "Super Realism" # Leave trigger_word blank if not used.
|
|
|
|
|
|
|
42 |
|
43 |
pipe.load_lora_weights(lora_repo)
|
44 |
pipe.to("cuda")
|