fffiloni commited on
Commit
8ca9d5e
·
verified ·
1 Parent(s): 7aa712c

enable xformers memory

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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,