Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -39,8 +39,8 @@ torch.backends.cuda.matmul.allow_fp16_reduced_precision_reduction = False
|
|
| 39 |
torch.backends.cudnn.allow_tf32 = False
|
| 40 |
torch.backends.cudnn.deterministic = False
|
| 41 |
torch.backends.cudnn.benchmark = False
|
| 42 |
-
|
| 43 |
-
|
| 44 |
torch.set_float32_matmul_precision("highest")
|
| 45 |
|
| 46 |
FTP_HOST = 'noahcohn.com'
|
|
@@ -123,7 +123,7 @@ def load_and_prepare_model():
|
|
| 123 |
#torch_dtype=torch.bfloat16,
|
| 124 |
add_watermarker=False,
|
| 125 |
# custom_pipeline="lpw_stable_diffusion_xl",
|
| 126 |
-
|
| 127 |
token=HF_TOKEN,
|
| 128 |
text_encoder=None,
|
| 129 |
text_encoder_2=None,
|
|
@@ -601,14 +601,14 @@ with gr.Blocks(theme=gr.themes.Origin(),css=css) as demo:
|
|
| 601 |
minimum=448,
|
| 602 |
maximum=MAX_IMAGE_SIZE,
|
| 603 |
step=64,
|
| 604 |
-
value=
|
| 605 |
)
|
| 606 |
height = gr.Slider(
|
| 607 |
label="Height",
|
| 608 |
minimum=448,
|
| 609 |
maximum=MAX_IMAGE_SIZE,
|
| 610 |
step=64,
|
| 611 |
-
value=
|
| 612 |
)
|
| 613 |
with gr.Row():
|
| 614 |
guidance_scale = gr.Slider(
|
|
@@ -616,14 +616,14 @@ with gr.Blocks(theme=gr.themes.Origin(),css=css) as demo:
|
|
| 616 |
minimum=0.1,
|
| 617 |
maximum=30,
|
| 618 |
step=0.1,
|
| 619 |
-
value=3.
|
| 620 |
)
|
| 621 |
num_inference_steps = gr.Slider(
|
| 622 |
label="Number of inference steps",
|
| 623 |
minimum=10,
|
| 624 |
maximum=1000,
|
| 625 |
step=10,
|
| 626 |
-
value=
|
| 627 |
)
|
| 628 |
|
| 629 |
gr.Examples(
|
|
|
|
| 39 |
torch.backends.cudnn.allow_tf32 = False
|
| 40 |
torch.backends.cudnn.deterministic = False
|
| 41 |
torch.backends.cudnn.benchmark = False
|
| 42 |
+
torch.backends.cuda.preferred_blas_library="cublas"
|
| 43 |
+
torch.backends.cuda.preferred_linalg_library="cusolver"
|
| 44 |
torch.set_float32_matmul_precision("highest")
|
| 45 |
|
| 46 |
FTP_HOST = 'noahcohn.com'
|
|
|
|
| 123 |
#torch_dtype=torch.bfloat16,
|
| 124 |
add_watermarker=False,
|
| 125 |
# custom_pipeline="lpw_stable_diffusion_xl",
|
| 126 |
+
use_safetensors=True,
|
| 127 |
token=HF_TOKEN,
|
| 128 |
text_encoder=None,
|
| 129 |
text_encoder_2=None,
|
|
|
|
| 601 |
minimum=448,
|
| 602 |
maximum=MAX_IMAGE_SIZE,
|
| 603 |
step=64,
|
| 604 |
+
value=1024,
|
| 605 |
)
|
| 606 |
height = gr.Slider(
|
| 607 |
label="Height",
|
| 608 |
minimum=448,
|
| 609 |
maximum=MAX_IMAGE_SIZE,
|
| 610 |
step=64,
|
| 611 |
+
value=1024,
|
| 612 |
)
|
| 613 |
with gr.Row():
|
| 614 |
guidance_scale = gr.Slider(
|
|
|
|
| 616 |
minimum=0.1,
|
| 617 |
maximum=30,
|
| 618 |
step=0.1,
|
| 619 |
+
value=3.75,
|
| 620 |
)
|
| 621 |
num_inference_steps = gr.Slider(
|
| 622 |
label="Number of inference steps",
|
| 623 |
minimum=10,
|
| 624 |
maximum=1000,
|
| 625 |
step=10,
|
| 626 |
+
value=200,
|
| 627 |
)
|
| 628 |
|
| 629 |
gr.Examples(
|