ford442 commited on
Commit
172acd9
·
1 Parent(s): c1965e2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -106,8 +106,8 @@ def apply_style(style_name: str, positive: str, negative: str = "") -> Tuple[str
106
  def load_and_prepare_model(model_id):
107
  model_dtypes = {"ford442/RealVisXL_V5.0_BF16": torch.bfloat16,}
108
  dtype = model_dtypes.get(model_id, torch.bfloat16) # Default to float32 if not found
109
- #vae = AutoencoderKL.from_pretrained("ford442/sdxl-vae-bf16",use_safetensors=True,safety_checker=None)
110
- vae = AutoencoderKL.from_pretrained("stabilityai/sdxl-vae",use_safetensors=True,safety_checker=None)
111
  # vae = AutoencoderKL.from_pretrained("BeastHF/MyBack_SDXL_Juggernaut_XL_VAE/MyBack_SDXL_Juggernaut_XL_VAE_V10(version_X).safetensors",safety_checker=None).to(torch.bfloat16)
112
  # vae = AutoencoderKL.from_pretrained("ford442/sdxl-vae-bf16", safety_checker=None).to('cuda')
113
  pipeX = StableDiffusionXLPipeline.from_pretrained("SG161222/RealVisXL_V5.0",use_safetensors=True)
@@ -458,14 +458,14 @@ with gr.Blocks(theme=gr.themes.Origin(),css=css) as demo:
458
  minimum=448,
459
  maximum=MAX_IMAGE_SIZE,
460
  step=64,
461
- value=1024,
462
  )
463
  height = gr.Slider(
464
  label="Height",
465
  minimum=448,
466
  maximum=MAX_IMAGE_SIZE,
467
  step=64,
468
- value=1024,
469
  )
470
  with gr.Row():
471
  guidance_scale = gr.Slider(
@@ -480,7 +480,7 @@ with gr.Blocks(theme=gr.themes.Origin(),css=css) as demo:
480
  minimum=10,
481
  maximum=1000,
482
  step=10,
483
- value=140,
484
  )
485
 
486
  gr.Examples(
 
106
  def load_and_prepare_model(model_id):
107
  model_dtypes = {"ford442/RealVisXL_V5.0_BF16": torch.bfloat16,}
108
  dtype = model_dtypes.get(model_id, torch.bfloat16) # Default to float32 if not found
109
+ vae = AutoencoderKL.from_pretrained("ford442/sdxl-vae-bf16",use_safetensors=True,safety_checker=None)
110
+ #vae = AutoencoderKL.from_pretrained("stabilityai/sdxl-vae",use_safetensors=True,safety_checker=None)
111
  # vae = AutoencoderKL.from_pretrained("BeastHF/MyBack_SDXL_Juggernaut_XL_VAE/MyBack_SDXL_Juggernaut_XL_VAE_V10(version_X).safetensors",safety_checker=None).to(torch.bfloat16)
112
  # vae = AutoencoderKL.from_pretrained("ford442/sdxl-vae-bf16", safety_checker=None).to('cuda')
113
  pipeX = StableDiffusionXLPipeline.from_pretrained("SG161222/RealVisXL_V5.0",use_safetensors=True)
 
458
  minimum=448,
459
  maximum=MAX_IMAGE_SIZE,
460
  step=64,
461
+ value=768,
462
  )
463
  height = gr.Slider(
464
  label="Height",
465
  minimum=448,
466
  maximum=MAX_IMAGE_SIZE,
467
  step=64,
468
+ value=768,
469
  )
470
  with gr.Row():
471
  guidance_scale = gr.Slider(
 
480
  minimum=10,
481
  maximum=1000,
482
  step=10,
483
+ value=150,
484
  )
485
 
486
  gr.Examples(