Upload 2 files

mod.py

@@ -74,10 +74,13 @@ def is_repo_exists(repo_id):
 
 
 def clear_cache():
-    torch.cuda.empty_cache()
-    torch.cuda.reset_max_memory_allocated()
-    torch.cuda.reset_peak_memory_stats()
-    gc.collect()
+    try:
+        torch.cuda.empty_cache()
+        torch.cuda.reset_max_memory_allocated()
+        torch.cuda.reset_peak_memory_stats()
+        gc.collect()
+    except Exception as e:
+        print(e)
 
 
 def deselect_lora():
@@ -348,3 +351,4 @@ load_prompt_enhancer.zerogpu = True
 fuse_loras.zerogpu = True
 preprocess_image.zerogpu = True
 get_control_params.zerogpu = True
+clear_cache.zerogpu = True