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(
|