Spaces:
Paused
Paused
Update app.py
Browse files
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
|