Spaces:
Paused
Paused
Update inference.py
Browse files- inference.py +1 -1
inference.py
CHANGED
@@ -91,7 +91,7 @@ if __name__ == "__main__":
|
|
91 |
controlnet=controlnet, interpolater=interpolater, scheduler=scheduler,
|
92 |
)
|
93 |
pipe.enable_vae_slicing()
|
94 |
-
|
95 |
pipe.to(device)
|
96 |
|
97 |
generator = torch.Generator(device="cuda")
|
|
|
91 |
controlnet=controlnet, interpolater=interpolater, scheduler=scheduler,
|
92 |
)
|
93 |
pipe.enable_vae_slicing()
|
94 |
+
pipe.enable_xformers_memory_efficient_attention()
|
95 |
pipe.to(device)
|
96 |
|
97 |
generator = torch.Generator(device="cuda")
|