Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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 |
-
|
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=
|
462 |
)
|
463 |
height = gr.Slider(
|
464 |
label="Height",
|
465 |
minimum=448,
|
466 |
maximum=MAX_IMAGE_SIZE,
|
467 |
step=64,
|
468 |
-
value=
|
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=
|
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(
|