Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -116,15 +116,15 @@ def load_and_prepare_model():
|
|
116 |
# 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)
|
117 |
#sched = EulerAncestralDiscreteScheduler.from_pretrained("SG161222/RealVisXL_V5.0", subfolder='scheduler',beta_schedule="scaled_linear", steps_offset=1,timestep_spacing="trailing"))
|
118 |
#sched = EulerAncestralDiscreteScheduler.from_pretrained("SG161222/RealVisXL_V5.0", subfolder='scheduler', steps_offset=1,timestep_spacing="trailing")
|
119 |
-
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=True, token=HF_TOKEN)
|
120 |
#sched = EulerAncestralDiscreteScheduler.from_pretrained('ford442/RealVisXL_V5.0_BF16', subfolder='scheduler',beta_schedule="scaled_linear")
|
121 |
#pipeX = StableDiffusionXLPipeline.from_pretrained("SG161222/RealVisXL_V5.0").to(torch.bfloat16)
|
122 |
#pipeX = StableDiffusionXLPipeline.from_pretrained("ford442/Juggernaut-XI-v11-fp32",use_safetensors=True)
|
123 |
pipe = StableDiffusionXLPipeline.from_pretrained(
|
124 |
-
|
125 |
#'ford442/Juggernaut-XI-v11-fp32',
|
126 |
# 'SG161222/RealVisXL_V5.0',
|
127 |
-
'John6666/uber-realistic-porn-merge-xl-urpmxl-v3-sdxl',
|
128 |
#torch_dtype=torch.bfloat16,
|
129 |
add_watermarker=False,
|
130 |
# custom_pipeline="lpw_stable_diffusion_xl",
|
@@ -168,7 +168,7 @@ def load_and_prepare_model():
|
|
168 |
pipe.vae = vaeX #.to(torch.bfloat16)
|
169 |
#pipe.unet = unetX
|
170 |
|
171 |
-
|
172 |
#pipe.vae.do_rescale=False
|
173 |
#pipe.vae.do_convert_rgb=True
|
174 |
pipe.vae.vae_scale_factor=8
|
|
|
116 |
# 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)
|
117 |
#sched = EulerAncestralDiscreteScheduler.from_pretrained("SG161222/RealVisXL_V5.0", subfolder='scheduler',beta_schedule="scaled_linear", steps_offset=1,timestep_spacing="trailing"))
|
118 |
#sched = EulerAncestralDiscreteScheduler.from_pretrained("SG161222/RealVisXL_V5.0", subfolder='scheduler', steps_offset=1,timestep_spacing="trailing")
|
119 |
+
sched = EulerAncestralDiscreteScheduler.from_pretrained('ford442/RealVisXL_V5.0_BF16', subfolder='scheduler',beta_schedule="scaled_linear", token=HF_TOKEN) #, beta_start=0.00085, beta_end=0.012, steps_offset=1,use_karras_sigmas=True, token=HF_TOKEN)
|
120 |
#sched = EulerAncestralDiscreteScheduler.from_pretrained('ford442/RealVisXL_V5.0_BF16', subfolder='scheduler',beta_schedule="scaled_linear")
|
121 |
#pipeX = StableDiffusionXLPipeline.from_pretrained("SG161222/RealVisXL_V5.0").to(torch.bfloat16)
|
122 |
#pipeX = StableDiffusionXLPipeline.from_pretrained("ford442/Juggernaut-XI-v11-fp32",use_safetensors=True)
|
123 |
pipe = StableDiffusionXLPipeline.from_pretrained(
|
124 |
+
'ford442/RealVisXL_V5.0_BF16',
|
125 |
#'ford442/Juggernaut-XI-v11-fp32',
|
126 |
# 'SG161222/RealVisXL_V5.0',
|
127 |
+
#'John6666/uber-realistic-porn-merge-xl-urpmxl-v3-sdxl',
|
128 |
#torch_dtype=torch.bfloat16,
|
129 |
add_watermarker=False,
|
130 |
# custom_pipeline="lpw_stable_diffusion_xl",
|
|
|
168 |
pipe.vae = vaeX #.to(torch.bfloat16)
|
169 |
#pipe.unet = unetX
|
170 |
|
171 |
+
pipe.vae.do_resize=False
|
172 |
#pipe.vae.do_rescale=False
|
173 |
#pipe.vae.do_convert_rgb=True
|
174 |
pipe.vae.vae_scale_factor=8
|