1inkusFace commited on
Commit
d51a40b
·
verified ·
1 Parent(s): ae21228

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -106,7 +106,7 @@ def apply_style(style_name: str, positive: str, negative: str = "") -> Tuple[str
106
  negative = ""
107
  return p.replace("{prompt}", positive), n + negative
108
 
109
- unetX = UNet2DConditionModel.from_pretrained("'ford442/RealVisXL_V5.0_BF16", low_cpu_mem_usage=False, token=True) #.to(device).to(torch.bfloat16) #.to(device=device, dtype=torch.bfloat16)
110
 
111
  def load_and_prepare_model():
112
  #vae = AutoencoderKL.from_pretrained("ford442/sdxl-vae-bf16", safety_checker=None)
 
106
  negative = ""
107
  return p.replace("{prompt}", positive), n + negative
108
 
109
+ unetX = UNet2DConditionModel.from_pretrained('ford442/RealVisXL_V5.0_BF16', subfolder='unet', low_cpu_mem_usage=False, token=True) #.to(device).to(torch.bfloat16) #.to(device=device, dtype=torch.bfloat16)
110
 
111
  def load_and_prepare_model():
112
  #vae = AutoencoderKL.from_pretrained("ford442/sdxl-vae-bf16", safety_checker=None)