ford442 commited on
Commit
0c86826
·
verified ·
1 Parent(s): 8919834

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -91,13 +91,13 @@ def load_and_prepare_model():
91
  'ford442/RealVisXL_V5.0_BF16',
92
  #torch_dtype=torch.bfloat16,
93
  token=HF_TOKEN,
94
- low_cpu_mem_usage = False,
95
  add_watermarker=False,
96
  )
97
  pipe.vae = vaeXL #.to(torch.bfloat16)
98
  pipe.scheduler = sched
99
- pipe.vae.do_resize=False
100
- pipe.vae.vae_scale_factor=8
101
  #pipe.to(device=device, dtype=torch.bfloat16)
102
  pipe.to(device)
103
  pipe.to(torch.bfloat16)
 
91
  'ford442/RealVisXL_V5.0_BF16',
92
  #torch_dtype=torch.bfloat16,
93
  token=HF_TOKEN,
94
+ # low_cpu_mem_usage = False,
95
  add_watermarker=False,
96
  )
97
  pipe.vae = vaeXL #.to(torch.bfloat16)
98
  pipe.scheduler = sched
99
+ #pipe.vae.do_resize=False
100
+ #pipe.vae.vae_scale_factor=8
101
  #pipe.to(device=device, dtype=torch.bfloat16)
102
  pipe.to(device)
103
  pipe.to(torch.bfloat16)