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,
|