Update app.py

Dartvauder · Nov 7, 2024 · ae4f75a · ae4f75a
1 parent 30e214a
commit ae4f75a
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/LaunchFile/app.py b/LaunchFile/app.py
@@ -327,6 +327,8 @@ def flush():
     gc.collect()
     if torch.cuda.is_available():
         torch.cuda.empty_cache()
+    if torch.backends.mps.is_available():
+        torch.mps.empty_cache()
 
 
 def stop_generation():