Spaces:
Running
on
Zero
Running
on
Zero
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")
|
@@ -93,7 +93,7 @@ def generate(
|
|
93 |
width=width,
|
94 |
height=height,
|
95 |
guidance_scale=guidance_scale,
|
96 |
-
num_inference_steps=
|
97 |
num_images_per_prompt=1,
|
98 |
output_type="pil",
|
99 |
).images
|
@@ -186,7 +186,7 @@ with gr.Blocks(css=css, theme="bethecloud/storj_theme") as demo:
|
|
186 |
minimum=1,
|
187 |
maximum=40,
|
188 |
step=1,
|
189 |
-
value=
|
190 |
)
|
191 |
|
192 |
style_selection = gr.Radio(
|
|
|
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/Castor-Gta6-Theme-Flux-LoRA"
|
41 |
+
trigger_word = "GTA 6 Theme, World of GTA 6 " # Leave trigger_word blank if not used.
|
42 |
pipe.load_lora_weights(lora_repo)
|
43 |
|
44 |
pipe.to("cuda")
|
|
|
93 |
width=width,
|
94 |
height=height,
|
95 |
guidance_scale=guidance_scale,
|
96 |
+
num_inference_steps=28,
|
97 |
num_images_per_prompt=1,
|
98 |
output_type="pil",
|
99 |
).images
|
|
|
186 |
minimum=1,
|
187 |
maximum=40,
|
188 |
step=1,
|
189 |
+
value=28,
|
190 |
)
|
191 |
|
192 |
style_selection = gr.Radio(
|