alexnasa commited on
Commit
f8bb5a3
·
verified ·
1 Parent(s): 7ec8b26

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -353,6 +353,7 @@ def generate_video(
353
  video_context=video_context,
354
  audio_context=audio_context,
355
  )
 
356
  print("successful generation")
357
 
358
  return str(output_path), current_seed
 
353
  video_context=video_context,
354
  audio_context=audio_context,
355
  )
356
+ torch.cuda.empty_cache()
357
  print("successful generation")
358
 
359
  return str(output_path), current_seed