prithivMLmods commited on
Commit
ea990b4
·
verified ·
1 Parent(s): 6fe9765

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -26,8 +26,8 @@ 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/Belle"
30
- trigger_word = "Belle the female character" # Leave trigger_word blank if not used.
31
 
32
  pipe.load_lora_weights(lora_repo)
33
  pipe.to("cuda")
@@ -82,7 +82,7 @@ def generate(
82
  width=width,
83
  height=height,
84
  guidance_scale=guidance_scale,
85
- num_inference_steps=28,
86
  num_images_per_prompt=1,
87
  output_type="pil",
88
  ).images
@@ -164,7 +164,7 @@ with gr.Blocks(css=css, theme="bethecloud/storj_theme") as demo:
164
  minimum=1,
165
  maximum=40,
166
  step=1,
167
- value=28,
168
  )
169
 
170
  style_selection = gr.Radio(
 
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/3d-Station-Toon"
30
+ trigger_word = "3d station toon" # Leave trigger_word blank if not used.
31
 
32
  pipe.load_lora_weights(lora_repo)
33
  pipe.to("cuda")
 
82
  width=width,
83
  height=height,
84
  guidance_scale=guidance_scale,
85
+ num_inference_steps=30,
86
  num_images_per_prompt=1,
87
  output_type="pil",
88
  ).images
 
164
  minimum=1,
165
  maximum=40,
166
  step=1,
167
+ value=30,
168
  )
169
 
170
  style_selection = gr.Radio(