Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -26,8 +26,11 @@ if not torch.cuda.is_available():
|
|
26 |
base_model = "black-forest-labs/FLUX.1-dev"
|
27 |
pipe = DiffusionPipeline.from_pretrained(base_model, torch_dtype=torch.bfloat16)
|
28 |
|
29 |
-
lora_repo = "strangerzonehf/Flux-Super-Realism-LoRA"
|
30 |
-
trigger_word = "Super Realism" # Leave trigger_word blank if not used.
|
|
|
|
|
|
|
31 |
|
32 |
pipe.load_lora_weights(lora_repo)
|
33 |
pipe.to("cuda")
|
@@ -141,14 +144,14 @@ with gr.Blocks(css=css, theme="bethecloud/storj_theme") as demo:
|
|
141 |
minimum=512,
|
142 |
maximum=2048,
|
143 |
step=64,
|
144 |
-
value=
|
145 |
)
|
146 |
height = gr.Slider(
|
147 |
label="Height",
|
148 |
minimum=512,
|
149 |
maximum=2048,
|
150 |
step=64,
|
151 |
-
value=
|
152 |
)
|
153 |
|
154 |
with gr.Row():
|
|
|
26 |
base_model = "black-forest-labs/FLUX.1-dev"
|
27 |
pipe = DiffusionPipeline.from_pretrained(base_model, torch_dtype=torch.bfloat16)
|
28 |
|
29 |
+
#lora_repo = "strangerzonehf/Flux-Super-Realism-LoRA"
|
30 |
+
#trigger_word = "Super Realism" # Leave trigger_word blank if not used.
|
31 |
+
|
32 |
+
lora_repo = "strangerzonehf/New-Journey-Art-Flux"
|
33 |
+
trigger_word = "New Journey Art" # Leave trigger_word blank if not used.
|
34 |
|
35 |
pipe.load_lora_weights(lora_repo)
|
36 |
pipe.to("cuda")
|
|
|
144 |
minimum=512,
|
145 |
maximum=2048,
|
146 |
step=64,
|
147 |
+
value=1280,
|
148 |
)
|
149 |
height = gr.Slider(
|
150 |
label="Height",
|
151 |
minimum=512,
|
152 |
maximum=2048,
|
153 |
step=64,
|
154 |
+
value=832,
|
155 |
)
|
156 |
|
157 |
with gr.Row():
|