Spaces:
Running
on
Zero
Running
on
Zero
enable xformers memory
Browse files
app.py
CHANGED
@@ -142,10 +142,10 @@ def main(video_path, audio_path, progress=gr.Progress(track_tqdm=True)):
|
|
142 |
unet = unet.to(dtype=torch.float16)
|
143 |
|
144 |
# set xformers
|
145 |
-
|
146 |
if is_xformers_available():
|
147 |
unet.enable_xformers_memory_efficient_attention()
|
148 |
-
|
149 |
|
150 |
pipeline = LipsyncPipeline(
|
151 |
vae=vae,
|
|
|
142 |
unet = unet.to(dtype=torch.float16)
|
143 |
|
144 |
# set xformers
|
145 |
+
|
146 |
if is_xformers_available():
|
147 |
unet.enable_xformers_memory_efficient_attention()
|
148 |
+
|
149 |
|
150 |
pipeline = LipsyncPipeline(
|
151 |
vae=vae,
|