Krebzonide commited on
Commit
b85fd37
·
1 Parent(s): 58927e7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -42,8 +42,8 @@ def generate(prompt, neg_prompt, samp_steps, guide_scale, lora_scale, progress=g
42
  num_inference_steps=samp_steps,
43
  guidance_scale=guide_scale,
44
  #cross_attention_kwargs={"scale": lora_scale},
45
- num_images_per_prompt=1,
46
- width=768,
47
  #generator=torch.manual_seed(97),
48
  ).images
49
  return [(img, f"Image {i+1}") for i, img in enumerate(images)]
@@ -58,7 +58,7 @@ with gr.Blocks(css=css) as demo:
58
  with gr.Row():
59
  samp_steps = gr.Slider(1, 100, value=25, step=1, label="Sampling steps")
60
  guide_scale = gr.Slider(1, 10, value=6, step=0.5, label="Guidance scale")
61
- lora_scale = gr.Slider(0, 1, value=0.5, step=0.01, label="LoRA power")
62
 
63
  submit_btn.click(generate, [prompt, negative_prompt, samp_steps, guide_scale, lora_scale], [gallery], queue=True)
64
 
 
42
  num_inference_steps=samp_steps,
43
  guidance_scale=guide_scale,
44
  #cross_attention_kwargs={"scale": lora_scale},
45
+ num_images_per_prompt=lora_scale,
46
+ width=600,
47
  #generator=torch.manual_seed(97),
48
  ).images
49
  return [(img, f"Image {i+1}") for i, img in enumerate(images)]
 
58
  with gr.Row():
59
  samp_steps = gr.Slider(1, 100, value=25, step=1, label="Sampling steps")
60
  guide_scale = gr.Slider(1, 10, value=6, step=0.5, label="Guidance scale")
61
+ lora_scale = gr.Slider(1, 6, value=1, step=1, label="LoRA power")
62
 
63
  submit_btn.click(generate, [prompt, negative_prompt, samp_steps, guide_scale, lora_scale], [gallery], queue=True)
64