prithivMLmods commited on
Commit
05b8319
·
verified ·
1 Parent(s): 3e2ae9e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
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/Canopus-LoRA-Flux-UltraRealism-2.0"
41
- trigger_word = "Ultra realistic" # Leave trigger_word blank if not used.
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=16,
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=23,
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(