Update app.py

Dartvauder · Oct 27, 2024 · d3fe13c · d3fe13c
1 parent c03530b
commit d3fe13c
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/LaunchFile/app.py b/LaunchFile/app.py
@@ -302,7 +302,8 @@ def print_system_info():
 
 def flush():
     gc.collect()
-    torch.cuda.empty_cache()
+    if torch.cuda.is_available():
+        torch.cuda.empty_cache()
 
 
 def stop_generation():