Spaces:
Running
on
Zero
Running
on
Zero
prithivMLmods
commited on
Commit
•
721a33b
1
Parent(s):
fbde1a7
Update app.py
Browse files
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.
|
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":
|
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=
|
529 |
step=1,
|
530 |
-
value=
|
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(
|