Update app.py
Browse files
app.py
CHANGED
@@ -107,12 +107,13 @@ def apply_style(style_name: str, positive: str, negative: str = "") -> Tuple[str
|
|
107 |
def load_and_prepare_model():
|
108 |
#vae = AutoencoderKL.from_pretrained("ford442/sdxl-vae-bf16", safety_checker=None)
|
109 |
vaeX = AutoencoderKL.from_pretrained("stabilityai/sdxl-vae", safety_checker=None, use_safetensors=False, low_cpu_mem_usage=False, torch_dtype=torch.float32, token=True) #.to(device).to(torch.bfloat16) #.to(device=device, dtype=torch.bfloat16)
|
|
|
110 |
pipe = StableDiffusionXLPipeline.from_pretrained(
|
111 |
-
'ford442/RealVisXL_V5.0_BF16',
|
112 |
#'ford442/Juggernaut-XI-v11-fp32',
|
113 |
# 'SG161222/RealVisXL_V5.0',
|
114 |
#'John6666/uber-realistic-porn-merge-xl-urpmxl-v3-sdxl',
|
115 |
-
|
116 |
add_watermarker=False,
|
117 |
# custom_pipeline="lpw_stable_diffusion_xl",
|
118 |
#use_safetensors=True,
|
@@ -121,7 +122,6 @@ def load_and_prepare_model():
|
|
121 |
text_encoder_2=None,
|
122 |
vae=None,
|
123 |
)
|
124 |
-
|
125 |
pipe.vae=vaeX
|
126 |
pipe.to(device=device, dtype=torch.bfloat16)
|
127 |
pipe.vae.set_default_attn_processor()
|
|
|
107 |
def load_and_prepare_model():
|
108 |
#vae = AutoencoderKL.from_pretrained("ford442/sdxl-vae-bf16", safety_checker=None)
|
109 |
vaeX = AutoencoderKL.from_pretrained("stabilityai/sdxl-vae", safety_checker=None, use_safetensors=False, low_cpu_mem_usage=False, torch_dtype=torch.float32, token=True) #.to(device).to(torch.bfloat16) #.to(device=device, dtype=torch.bfloat16)
|
110 |
+
sched = EulerAncestralDiscreteScheduler.from_pretrained('ford442/RealVisXL_V5.0_BF16', subfolder='scheduler',beta_schedule="scaled_linear", beta_start=0.00085, beta_end=0.012, steps_offset=1,use_karras_sigmas=False)
|
111 |
pipe = StableDiffusionXLPipeline.from_pretrained(
|
112 |
+
#'ford442/RealVisXL_V5.0_BF16',
|
113 |
#'ford442/Juggernaut-XI-v11-fp32',
|
114 |
# 'SG161222/RealVisXL_V5.0',
|
115 |
#'John6666/uber-realistic-porn-merge-xl-urpmxl-v3-sdxl',
|
116 |
+
'John6666/uber-realistic-porn-merge-xl-urpmxl-v6final-sdxl',
|
117 |
add_watermarker=False,
|
118 |
# custom_pipeline="lpw_stable_diffusion_xl",
|
119 |
#use_safetensors=True,
|
|
|
122 |
text_encoder_2=None,
|
123 |
vae=None,
|
124 |
)
|
|
|
125 |
pipe.vae=vaeX
|
126 |
pipe.to(device=device, dtype=torch.bfloat16)
|
127 |
pipe.vae.set_default_attn_processor()
|