ford442 commited on
Commit
08d87ef
·
verified ·
1 Parent(s): eb8fd07

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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
- #torch_dtype=torch.bfloat16,
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()