Spaces:
Running
on
A10G
Running
on
A10G
rynmurdock
commited on
Commit
•
b3aaea2
1
Parent(s):
1913873
__pycache__/safety_checker_improved.cpython-310.pyc
CHANGED
Binary files a/__pycache__/safety_checker_improved.cpython-310.pyc and b/__pycache__/safety_checker_improved.cpython-310.pyc differ
|
|
app.py
CHANGED
@@ -49,7 +49,7 @@ pipe.load_ip_adapter("h94/IP-Adapter", subfolder="sdxl_models", weight_name="ip-
|
|
49 |
pipe.register_modules(image_encoder = image_encoder)
|
50 |
pipe.set_ip_adapter_scale(0.8)
|
51 |
|
52 |
-
pipe.vae = AutoencoderTiny.from_pretrained("madebyollin/taesdxl", torch_dtype=torch.
|
53 |
pipe.scheduler = EulerDiscreteScheduler.from_config(pipe.scheduler.config, timestep_spacing="trailing")
|
54 |
|
55 |
pipe.to(device=DEVICE)
|
|
|
49 |
pipe.register_modules(image_encoder = image_encoder)
|
50 |
pipe.set_ip_adapter_scale(0.8)
|
51 |
|
52 |
+
pipe.vae = AutoencoderTiny.from_pretrained("madebyollin/taesdxl", torch_dtype=torch.float16, low_cpu_mem_usage=True)
|
53 |
pipe.scheduler = EulerDiscreteScheduler.from_config(pipe.scheduler.config, timestep_spacing="trailing")
|
54 |
|
55 |
pipe.to(device=DEVICE)
|