prithivMLmods commited on
Commit
721a33b
1 Parent(s): fbde1a7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -286,7 +286,7 @@ device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
286
  #Compile
287
  if torch.cuda.is_available():
288
  pipe = StableDiffusionXLPipeline.from_pretrained(
289
- "SG161222/RealVisXL_V5.0", #(or) use --- SG161222/RealVisXL_V4.0 / SG161222/RealVisXL_V4.0_Lightning --- for better results.
290
  torch_dtype=torch.float16,
291
  use_safetensors=True,
292
  add_watermarker=False,
@@ -394,7 +394,7 @@ def generate(
394
  "width": width,
395
  "height": height,
396
  "guidance_scale": guidance_scale,
397
- "num_inference_steps": 20,
398
  "generator": generator,
399
  "num_images_per_prompt": num_images,
400
  "use_resolution_binning": use_resolution_binning,
@@ -525,9 +525,9 @@ with gr.Blocks(css=css, theme="bethecloud/storj_theme") as demo:
525
  num_inference_steps = gr.Slider(
526
  label="Steps",
527
  minimum=10,
528
- maximum=30,
529
  step=1,
530
- value=15,
531
  )
532
  with gr.Row():
533
  num_images_per_prompt = gr.Slider(
 
286
  #Compile
287
  if torch.cuda.is_available():
288
  pipe = StableDiffusionXLPipeline.from_pretrained(
289
+ "SG161222/RealVisXL_V5.0_Lightning", #(or) use --- SG161222/RealVisXL_V4.0 / SG161222/RealVisXL_V4.0_Lightning --- for better results.
290
  torch_dtype=torch.float16,
291
  use_safetensors=True,
292
  add_watermarker=False,
 
394
  "width": width,
395
  "height": height,
396
  "guidance_scale": guidance_scale,
397
+ "num_inference_steps": 30,
398
  "generator": generator,
399
  "num_images_per_prompt": num_images,
400
  "use_resolution_binning": use_resolution_binning,
 
525
  num_inference_steps = gr.Slider(
526
  label="Steps",
527
  minimum=10,
528
+ maximum=60,
529
  step=1,
530
+ value=30,
531
  )
532
  with gr.Row():
533
  num_images_per_prompt = gr.Slider(