Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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 |
-
|
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)
|