ford442 commited on
Commit
cac73c6
·
verified ·
1 Parent(s): 269f36c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -7
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
- #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,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
- #use_safetensors=True,
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=768,
605
  )
606
  height = gr.Slider(
607
  label="Height",
608
  minimum=448,
609
  maximum=MAX_IMAGE_SIZE,
610
  step=64,
611
- value=768,
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.8,
620
  )
621
  num_inference_steps = gr.Slider(
622
  label="Number of inference steps",
623
  minimum=10,
624
  maximum=1000,
625
  step=10,
626
- value=170,
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(